Expose gpus SLURM parameter

[tcompa]

Ref https://docs.s3it.uzh.ch/cluster/python_gpu_example/#creating-an-environment-for-tensorflow-on-a-gpu-node:

srun --pty -n 1 -c 2 --time=01:00:00 --gpus=1 --mem=8G bash -l