Last active
October 24, 2023 19:54
-
-
Save kchawla-pi/08d759028dc310d9cec0fabe98f0ccb4 to your computer and use it in GitHub Desktop.
Making GPUs available in Docker on Ubuntu 22.04
This file contains hidden or bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
sudo apt update | |
sudo add-apt-repository ppa:graphics-drivers/ppa | |
sudo apt install nvidia-driver-535-server | |
sudo apt install linux-modules-nvidia-535-server-generic | |
sudo apt-cache policy linux-modules-nvidia-535-server-$(uname -r) | |
sudo apt install linux-modules-nvidia-535-server-generic | |
sudo reboot | |
sudo apt update | |
sudo lsmod | grep nouveau | |
sudo lsmod | grep nvidia | |
nvidia-smi | |
sudo ubuntu-drivers devices | |
sudo apt install nvidia-cuda-toolkit --install-recommends | |
nvidia-smi | |
nvcc -V | |
printenv | |
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/deb/nvidia-container-toolkit.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 | |
sudo apt install nvidia-container-toolkit | |
sudo nvidia-ctk runtime configure --runtime=docker | |
sudo systemctl restart docker |
Sign up for free
to join this conversation on GitHub.
Already have an account?
Sign in to comment