diff --git a/README.md b/README.md index 51648c7addccc3ecf40e01bf5418d7643d7df258..ac13bbd2649a8d8c26e9bcde1a753ed8f9f252fd 100644 --- a/README.md +++ b/README.md @@ -6,7 +6,7 @@ Please read Enroot's [github page](https://github.com/NVIDIA/enroot) for more in ## Install enroot -**Enroot has been installed on Berzelius. You can skip this step if you plan to use it on Berzeliu**. +Enroot has been installed on Berzelius. You can skip this step if you plan to use it on Berzeliu. - For Debian-based distributions ``` @@ -18,6 +18,22 @@ sudo apt install -y ./*.deb - For others, see [here](https://github.com/NVIDIA/enroot/blob/master/doc/installation.md) +## GPU support using libnvidia-container + +Enroot comes with GPU support using libnvidia-container. Enroot has been installed on Berzelius. You can skip this step if you plan to use it on Berzeliu. + +To install libnvidia-container on your local machine, see the [instructions](https://nvidia.github.io/libnvidia-container/). + +- For Debian-based distributions +``` +distribution=$(. /etc/os-release;echo $ID$VERSION_ID) \ + && curl -fsSL https://nvidia.github.io/libnvidia-container/gpgkey | sudo gpg --dearmor -o /usr/share/keyrings/nvidia-container-toolkit-keyring.gpg \ + && curl -s -L https://nvidia.github.io/libnvidia-container/$distribution/libnvidia-container.list | \ + sed 's#deb https://#deb [signed-by=/usr/share/keyrings/nvidia-container-toolkit-keyring.gpg] https://#g' | \ + sudo tee /etc/apt/sources.list.d/nvidia-container-toolkit.list +sudo apt-get update +``` + ## Set up Nvidia credentials This step is necessary for importing container images from [Nvidia NGC](https://catalog.ngc.nvidia.com/containers).