ubuntu 安装nvida-docker2

前提是你已经安装了docker

# 配置源
distribution=$(. /etc/os-release;echo $ID$VERSION_ID) \
   && curl -s -L https://nvidia.github.io/nvidia-docker/gpgkey | sudo apt-key add - \
   && curl -s -L https://nvidia.github.io/nvidia-docker/$distribution/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 systemctl restart docker
sudo docker run --rm --gpus all nvidia/cuda:[你的cuda版本]-base nvidia-smi

检查了一下，我当前机器的显卡核心是384的 ，看了一下官网说明，384需要装9.0版本的驱动
我的就是

sudo docker run --rm --gpus all nvidia/cuda:9.0-base nvidia-smi

https://github.com/NVIDIA/nvidia-docker/wiki/CUDA#requirements

CUDA toolkit version	Driver version	GPU architecture
6.5	>= 340.29	>= 2.0 (Fermi)
7.0	>= 346.46	>= 2.0 (Fermi)
7.5	>= 352.39	>= 2.0 (Fermi)
8.0	== 361.93 or >= 375.51	== 6.0 (P100)
8.0	>= 367.48	>= 2.0 (Fermi)
9.0	>= 384.81	>= 3.0 (Kepler)
9.1	>= 387.26	>= 3.0 (Kepler)
9.2	>= 396.26	>= 3.0 (Kepler)
10.0	>= 384.130, < 385.00	Tesla GPUs
10.0	>= 410.48	>= 3.0 (Kepler)