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