[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