Skip to content

Instantly share code, notes, and snippets.

@hiropppe
Last active June 11, 2018 04:58
Show Gist options
  • Save hiropppe/d9421a7dddbb47a47edb776b1051707d to your computer and use it in GitHub Desktop.
Save hiropppe/d9421a7dddbb47a47edb776b1051707d to your computer and use it in GitHub Desktop.
nvidia-docker2 setup instruction for for Ubuntu 16.04 LTS (Xenial Xerus) on EC2 g3.4xlarge.
[2018.6.11]
3 sudo apt update
4 sudo apt upgrade
6 sudo apt install build-essential
11 wget https://developer.download.nvidia.com/compute/cuda/repos/ubuntu1604/x86_64/cuda-repo-ubuntu1604_9.2.88-1_amd64.deb
12 sudo dpkg -i cuda-repo-ubuntu1604_9.2.88-1_amd64.deb
13 sudo apt-key adv --fetch-keys http://developer.download.nvidia.com/compute/cuda/repos/ubuntu1604/x86_64/7fa2af80.pub
14 sudo apt-get update
15 sudo apt-get install cuda
16 nvidia-smi
17 sudo apt-get update
18 sudo apt-get install apt-transport-https ca-certificates curl software-properties-common
19 curl -fsSL https://download.docker.com/linux/ubuntu/gpg | sudo apt-key add -
20 sudo apt-key fingerprint 0EBFCD88
21 sudo add-apt-repository "deb [arch=amd64] https://download.docker.com/linux/ubuntu \
22 $(lsb_release -cs) \
23 stable"
24 sudo apt-get upate
25 sudo apt-get update
26 sudo apt-get install docker-ce
27 sudo docker run hello-world
28 sudo usermod -aG docker $USER
30 exit
31 docker run hello-world
32 curl -s -L https://nvidia.github.io/nvidia-docker/gpgkey | sudo apt-key add -
33 distribution=$(. /etc/os-release;echo $ID$VERSION_ID)
34 curl -s -L https://nvidia.github.io/nvidia-docker/$distribution/nvidia-docker.list | sudo tee /etc/apt/sources.list.d/nvidia-docker.list
35 sudo apt-get update
36 sudo apt-get install -y nvidia-docker2
37 sudo pkill -SIGHUP dockerd
38 docker run --runtime=nvidia --rm nvidia/cuda nvidia-smi
[2018.1]
sudo apt update
sudo apt upgrade
sudo apt install build-essential
sudo apt install dkms
wget https://developer.download.nvidia.com/compute/cuda/repos/ubuntu1604/x86_64/cuda-repo-ubuntu1604_8.0.61-1_amd64.deb
sudo dpkg -i cuda-repo-ubuntu1604_8.0.61-1_amd64.deb
sudo apt-get update
sudo apt-get install cuda
sudo apt-get install apt-transport-https ca-certificates curl software-properties-common
curl -fsSL https://download.docker.com/linux/ubuntu/gpg | sudo apt-key add -
sudo apt-key fingerprint 0EBFCD88
sudo add-apt-repository "deb [arch=amd64] https://download.docker.com/linux/ubuntu \
$(lsb_release -cs) \
stable"
sudo apt-get update
sudo apt-get install docker-ce
sudo docker run hello-world
sudo usermod -aG docker $USER
exit
# login
docker run hello-world
curl -s -L https://nvidia.github.io/nvidia-docker/gpgkey | sudo apt-key add -
curl -s -L https://nvidia.github.io/nvidia-docker/ubuntu16.04/amd64/nvidia-docker.list | sudo tee /etc/apt/sources.list.d/nvidia-docker.list
sudo apt-get update
sudo apt-get install -y nvidia-docker2
sudo pkill -SIGHUP dockerd
docker run --runtime=nvidia --rm nvidia/cuda nvidia-smi
docker run --runtime=nvidia --rm nvidia/cuda:8.0-cudnn6-devel-ubuntu16.04 nvidia-smi
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment