先决条件
运行nvidia-docker 2.0的先决条件列表如下所述。
- GNU/Linux x86_64 内核版本 > 3.10
- Docker >= 1.12
- NVIDIA GPU with Architecture > Fermi (2.1)
- NVIDIA drivers ~= 361.93 (旧版本未经测试)
你的驱动程序版本可能会限制你的CUDA功能（请参阅CUDA要求）
安装2.0版
首先，确保你安装了NVIDIA驱动和Docker支持的版本。
如果您有自定义的/etc/docker/daemon.json，则nvidia-docker2包可能会覆盖它。
安装时，你必须固定nvidia-docker2和nvidia-container-runtime的版本，例如：
sudo apt-get install -y nvidia-docker2=2.0.1+docker1.12.6-1 nvidia-container-runtime=1.1.0+docker1.12.6-1
使用 apt-cache madison nvidia-docker2 nvidia-container-runtime 或 yum search --showduplicates nvidia-docker2 nvidia-container-runtime 列出所有可用的版本。
实战操作如下：
## 我需要18.03版本的，所以我先查看我可用的版本
yum search --showduplicates nvidia-docker2 nvidia-container-runtime list|grep 18.03
安装
## 添加aliyun镜像库
yum-config-manager —add-repo http://mirrors.aliyun.com/docker-ce/linux/centos/docker-ce.repo
yum install -y nvidia-docker2-2.0.3-1.docker18.03.1.ce.noarch
yum install -y nvidia-container-runtime-2.0.0-1.docker18.03.1.x86_64
其他版本和体系结构
按照此处的说明分发安装存储库。
测试
nvidia-docker运行时向Docker守护程序注册一个新的容器，使用docker run时必须选择nvidia：
docker run --runtime=nvidia --rm nvidia/cuda nvidia-smi
