slurm: allow predictions on gpu

This commit is contained in:
Starbeamrainbowlabs 2022-09-23 19:21:57 +01:00
parent d59de41ebb
commit 58c65bdc86
Signed by: sbrl
GPG key ID: 1BE5172E637709C2

View file

@ -5,7 +5,7 @@
#SBATCH --gres=gpu:1
#SBATCH -o %j.%N.%a.out.log
#SBATCH -e %j.%N.%a.err.log
#SBATCH -p gpu05
#SBATCH -p gpu05,gpu
#SBATCH --time=5-00:00:00
module load utilities/multi