uw-psych / compute_docs

https://uw-psych.github.io/compute_docs/
Creative Commons Attribution 4.0 International
1 stars 3 forks source link

Run SLEAP batch jobs in hyak #8

Open arokem opened 1 year ago

arokem commented 1 year ago

Relevant documentation: https://sleap.ai/

Potentially use miniconda containers as a base to build this.

This set of instructions is useful for setting tensorflow up to see the GPU in conda/hyak: https://www.tensorflow.org/install/pip#step-by-step_instructions.

Need to add the gpu flag to calls to sbatch.