...
Code Block | ||
---|---|---|
| ||
#!/bin/bash #SBATCH --account=<account> # Your account #SBATCH --time 0:10:00 #SBATCH -p gpu #SBATCH --ntasks=1 #SBATCH --gpugpus=a100:2 # allocate 2 (out of 4) A100 GPUs; to get 2 (out of 2) A40 GPUs use --gpus=a40:2 #SBATCH --hint=nomultithread #SBATCH --job-name=gpu #SBATCH --output=out_%x.%j # disable hyperthreading #SBATCH --hint=nomultithread ## Uncomment the following line to enlarge the stacksize if needed, ## e.g., if your code crashes with a spurious segmentation fault. # ulimit -s unlimited # To be on the safe side, we emphasize that it is pure MPI, no OpenMP threads export OMP_NUM_THREADS=1 srun your_code_that_runs_on_GPUs |
...