...
Column |
---|
|
Code Block |
---|
language | bash |
---|
theme | Emacs |
---|
title | Listing 23. GPU job packing with --gpu-bind |
---|
| #!/bin/bash --login
#SBATCH --account=[your-account]
#SBATCH --partition=nvlinkq
#SBATCH --nodes=1
#SBATCH --ntasks=4
#SBATCH --ntasks-per-socket=2
#SBATCH --cpus-pert-task=1
#SBATCH --gres=gpu:4
#SBATCH --gpu-bind=map_gpu:0,1,2,3
#SBATCH --time=00:10:00
#SBATCH
--account=[your-account] #SBATCH
--export=NONE
#Default loaded compiler module is gcc module
module load cuda
#Run the cuda executable from a wrapper:
srun -u -N 1 -n 4 -c 1 wrapper.sh |
|
And the wrapper.sh
in this case is:
...