17 lines
401 B
Bash
17 lines
401 B
Bash
#!/bin/bash
|
|
|
|
srun -K \
|
|
--container-image=PATH/TO/SLURM/IMAGE \
|
|
--container-workdir="$(pwd)" \
|
|
--container-mounts=/ALL/IMPORTANT/MOUNTS,"$(pwd)":"$(pwd)" \
|
|
--partition=RTXA6000,RTX3090,A100-40GB,A100-80GB,H100,H200 \
|
|
--job-name="python" \
|
|
--nodes=1 \
|
|
--gpus=1 \
|
|
--ntasks=1 \
|
|
--cpus-per-task=24 \
|
|
--mem=64G \
|
|
--time=1-0 \
|
|
--export="NLTK_DATA=/PATH/TO/NLTK_DATA" \
|
|
python3 "$@"
|