Skip to content

Instantly share code, notes, and snippets.

@arcaduf

arcaduf/Non-interactive GPU session with Slurm

Last active September 6, 2017 12:18
Show Gist options
  • Save arcaduf/d8e4f58a4a27d2c8b22588b6a1e15ee7 to your computer and use it in GitHub Desktop.
Save arcaduf/d8e4f58a4a27d2c8b22588b6a1e15ee7 to your computer and use it in GitHub Desktop.
Download ZIP
Non-interactive GPU session with Slurm
Raw
Non-interactive GPU session with Slurm
virtualenv ~/mypio
source ~/mypio/bin/activate
pip install scipy scikit-image scikit-learn SimpleITK h5py
deactivate
sbatch shellwrapper.sh
Inside the shellwrapper:
#!/bin/bash
#SBATCH --partition gpu
#SBATCH --gres gpu:1
#SBATCH --mail-type=END,FAIL
#SBATCH [email protected]
ml python2
ml virtualenv
ml CUDA cuDNN
ml OpenCV/2.4.11-goolf-1.7.20
ml TensorFlow/1.1.0-goolf-1.7.20-CUDA-8.0.44-Python-2.7.11
source ~/mypio/bin/activate
<cmd>
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment