Ubuntu18.04安装NVIDIA Docker 2

背景：安装NVIDIA transfer，需要依赖环境。为了在docker中支持GPU，NVidia之前是弄了个nvidia-docker2，现在升级为NVIDIA Container Toolkit了。

参考github

---

安装

1. Docker安装传送门

完成上面的Docker安装后，开始配置nvidia-docker源

2. 添加源

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

3. 安装nvidia-container-toolkit并重启docker

sudo apt-get update && sudo apt-get install -y nvidia-container-toolkit
sudo systemctl restart docker

测试

1. 使用最新的CUDA官方图测试nvidia-smi

$ sudo docker run --gpus all nvidia/cuda:10.0-base nvidia-smi

2. 启动支持双GPU的容器

$ sudo docker run --gpus 2 nvidia/cuda:10.0-base nvidia-smi

3. 指定GPU 1，运行容器

$ sudo docker run --gpus 2 nvidia/cuda:9.0-base nvidia-smi

4. 指定个人容器性能（很少用）

$ sudo docker run --gpus all,capabilities=utility nvidia/cuda:10.0-base nvidia-smi