Last active
March 24, 2023 15:25
-
-
Save ebr/167eb13ef8a5276312680c73dfd3e25c to your computer and use it in GitHub Desktop.
quickstart docker + nvidia on ubuntu
This file contains 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
#!/bin/bash | |
sudo apt update | |
sudo apt-get install -y ca-certificates curl gnupg lsb-release | |
sudo mkdir -p /etc/apt/keyrings && curl -fsSL https://download.docker.com/linux/ubuntu/gpg | sudo gpg --dearmor -o /etc/apt/keyrings/docker.gpg | |
echo "deb [arch=$(dpkg --print-architecture) signed-by=/etc/apt/keyrings/docker.gpg] https://download.docker.com/linux/ubuntu $(lsb_release -cs) stable" | sudo tee /etc/apt/sources.list.d/docker.list > /dev/null | |
sudo apt-get install -y docker-ce docker-ce-cli containerd.io docker-compose-plugin | |
sudo usermod -aG docker ubuntu | |
cat <<EOF | sudo tee /etc/docker/config.json | |
{ | |
"features": { | |
"buildkit": true | |
}, | |
"runtimes": { | |
"nvidia": { | |
"path": "nvidia-container-runtime", | |
"runtimeArgs": [] | |
} | |
}, | |
"default-runtime": "nvidia" | |
} | |
EOF |
Sign up for free
to join this conversation on GitHub.
Already have an account?
Sign in to comment