Using NVIDIA enroot to run containers
Before using enroot¶
enroot will attempt to install containers in
$HOME. In most cases, it will be
necessary to place these files somewhere else to prevent hitting quota limts.
For example, they can be moved to the barn:
mkdir -p $HOME/.local/share ln -s $(readlink $HOME/barn)/enroot $HOME/.local/share/enroot
Importing container images¶
enroot can import container images and package a single squashfs file system used to create enroot containers.
CCI does not allow users the admin/root privileges necessary to import container images on CCI systems. Users wishing to import container images must do so on another system where they have root privileges. Instructions for installing and importing container images with enroot are available on the enroot github page.
Create refers to creating an instance of a container using a squashfs container image. Once a container image has been imported, a container can be created:
enroot create --name mycontainer previously-imported-image.sqsh
Once a container has been created, it can be run with enroot using sbatch or srun like any other application. Normal requirements for Slurm on each cluster apply such as specifying time limits or GPU requirements.
Although not strictly necessary, you may also wish to pass environment
variables to your containers. This can be done with the
enroot start -e http_proxy=http://proxy:8888 -e https_proxy=http://proxy:8888 entrypoint
srun enroot start entrypoint
Example batch script to run with
1 2 3
There are some example containers provided that can be created and used to sanity check an environment. They are based on the NVIDIA CUDA runtime containers and have the corresponding CUDA samples installed. The deviceQuery sample is already built so the container can be created and run to test the environment, ex:
enroot create --name mycontainer /gpfs/u/software/container-examples/cuda-11.1-runtime-samples.sqsh srun --gres=gpu:2 -t 5:00 enroot start mycontainer /usr/local/cuda/samples/1_Utilities/deviceQuery/deviceQuery
Examples provided in