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

-- 
slurm-users mailing list -- slurm-users@lists.schedmd.com
To unsubscribe send an email to slurm-users-le...@lists.schedmd.com

Reply via email to