diff --git a/DOCKER_SETUP.md b/DOCKER_SETUP.md index 4ccc741..80579b8 100644 --- a/DOCKER_SETUP.md +++ b/DOCKER_SETUP.md @@ -16,20 +16,22 @@ Before proceeding, ensure the following are installed and configured on your sys 3. **NVIDIA Container Toolkit** - Enables Docker to access the GPU resources on the host machine. - **Installation Steps:** - ```bash - # Add the package repositories + + ```bash + # Add the package repositories 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/stable/$distribution/nvidia-container-toolkit.list | \ - sudo tee /etc/apt/sources.list.d/nvidia-container-toolkit.list - - # Install the NVIDIA Container Toolkit - sudo apt-get update - sudo apt-get install -y nvidia-docker2 + && curl -s -L https://nvidia.github.io/libnvidia-container/stable/$distribution/nvidia-container-toolkit.list | \ + sudo tee /etc/apt/sources.list.d/nvidia-container-toolkit.list + + # Install the NVIDIA Container Toolkit + sudo apt-get update + sudo apt-get install -y nvidia-docker2 + + # Restart the Docker daemon to apply changes + sudo systemctl restart docker + ``` - # Restart the Docker daemon to apply changes - sudo systemctl restart docker - ``` - **Reference:** [NVIDIA Docker Documentation](https://docs.nvidia.com/datacenter/cloud-native/container-toolkit/install-guide.html) 4. **GitHub Repository Clone**