前提是你已经安装了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) |