[slurm-users] Re: [EXT] Re: Issue with Enforcing GPU Usage Limits in Slurm

Hi! Christ. Thank you again for your instruction. I've tried version 23.11.10. It does work. When I ran the script using the following command, it successfully restricted the usage to the specified CUDA devices: srun -p gpu --gres=gpu:2 -nodelist=node11 python test.py And when I checked the GPUs using this command, I saw the expected number of GPUs: srun -p gpu --gres=gpu:2 -nodelist=node11 --pty nvidia-smi Thank you very much for your guidance. Best luck Lyz