Skip to content

Instantly share code, notes, and snippets.

@stevenc81
Created February 26, 2018 01:38
Show Gist options
  • Save stevenc81/1fd1021a50fd6505e0963853f731ba6a to your computer and use it in GitHub Desktop.
Save stevenc81/1fd1021a50fd6505e0963853f731ba6a to your computer and use it in GitHub Desktop.
$ curl -s -L https://nvidia.github.io/nvidia-docker/gpgkey | \
sudo apt-key add -
$ curl -s -L https://nvidia.github.io/nvidia-docker/debian9/amd64/nvidia-docker.list | \
sudo tee /etc/apt/sources.list.d/nvidia-docker.list
$ sudo apt-get update && sudo apt-get install -y \
apt-transport-https \
ca-certificates \
curl \
gnupg2 \
software-properties-common
$ curl -fsSL https://download.docker.com/linux/$(. /etc/os-release; echo "$ID")/gpg | sudo apt-key add -
$ sudo apt-key fingerprint 0EBFCD88
$ sudo add-apt-repository \
"deb [arch=amd64] https://download.docker.com/linux/$(. /etc/os-release; echo "$ID") \
$(lsb_release -cs) \
stable"
$ sudo apt-get update && sudo apt-get install -y \
nvidia-docker2 \
docker-ce
$ sudo vim /lib/systemd/system/docker.service
ExecStart=/usr/bin/dockerd -H fd:// -s=overlay2
$ sudo systemctl daemon-reload
$ sudo apt-get update && sudo apt-get install -y \
nvidia-docker2 \
docker-ce
$ sudo tee /etc/docker/daemon.json <<EOF
{
"default-runtime": "nvidia",
"runtimes": {
"nvidia": {
"path": "/usr/bin/nvidia-container-runtime",
"runtimeArgs": []
}
}
}
EOF
$ sudo pkill -SIGHUP dockerd
$ sudo systemctl restart kubelet
// Verify
$ sudo docker run --rm nvidia/cuda nvidia-smi
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment