You are here: Foswiki>Main Web>SampleSGESubmitScripts>GPU (21 Apr 2020, AdminUser)Edit Attach

Sample GPU submit script using the Caffe training network

Using the Tesla K80 cards: 
#SBATCH --partition="gpu"
#SBATCH --nodes=1
#SBATCH --gres=gpu:k80:1

. /etc/profile.d/modules.sh

module load shared cuda90/toolkit caffe

cd /work/abc123/caffe_workdir
./examples/mnist/train_lenet.sh

For the V100 cards: 
#SBATCH --partition="gpu-v100"
#SBATCH --nodes=1
#SBATCH --gres=gpu:v100:1
 
. /etc/profile.d/modules.sh

module load shared cuda90/toolkit caffe

cd /work/abc123/caffe_workdir
./examples/mnist/train_lenet.sh

-- AdminUser - 14 Jun 2018
Edit  | Attach | Print version | History: r3 < r2 < r1 | Backlinks | View wiki text | Edit wiki text | More topic actions
Topic revision: r3 - 21 Apr 2020, AdminUser
This site is powered by FoswikiCopyright © by the contributing authors. All material on this collaboration platform is the property of the contributing authors.
Ideas, requests, problems regarding Foswiki? Send feedback