Skip to content

Instantly share code, notes, and snippets.

@kchawla-pi
Last active October 24, 2023 19:54
Show Gist options
  • Save kchawla-pi/08d759028dc310d9cec0fabe98f0ccb4 to your computer and use it in GitHub Desktop.
Save kchawla-pi/08d759028dc310d9cec0fabe98f0ccb4 to your computer and use it in GitHub Desktop.
Making GPUs available in Docker on Ubuntu 22.04
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