Containers

Support

Docker containers are supported via NVIDIA enroot with pyxis plugin (SLURM only). For Pyxis usage please refer to https://github.com/NVIDIA/pyxis#usage

Examples

Start an interactive job in Ubuntu container

srun -N1 -p hpc --exclusive --container-image=ubuntu --pty bash

Reusing a container within a job with `--container-name` switch

Allocate a node

salloc -N1 -p hpc --exclusive

Run an interactive session in Ubuntu container from Docker Hub

srun --container-image=ubuntu --container-name=ubuntu_dev --pty bash

When --container-name option is specified, the container will not be automatically removed after the slurm task is complete.

Install any packages inside the container

apt-get update
apt-get install gcc make

Exit the session and reuse the container using its name for another bash session

exit
srun --container-name=ubuntu_dev --pty bash
gcc --version
exit

Deploy NVIDIA pytorch container from NGC repository and start an interactive job

srun -p ndv4 -N 1 --exclusive --gpus-per-node=8 --container-image='nvcr.io#nvidia/pytorch:21.07-py3' --container-name=pytorch --pty bash