Accessing interactive GPU node:
- srun --export ALL --pty --x11 -p gpu bash
To request a GPU device from a GPU node, you need to provide GPU request options.
For the above example, it should be:
- srun --export ALL --pty --x11 -p gpu --gres=gpu:1 bash
In a SLURM script:
- #SBATCH --gres=gpu:1
For a SLURM command line,
- sbatch --partition=gpu --gres=gpu:1 --wrap="nvidia-smi"
Reference:
- https://slurm.schedmd.com/gres.html#Running_Jobs