ubuntu安装nvidia-docker2

首先前提是已经安装了nvidia驱动，未安装nvidia驱动的可以参考：https://blog.csdn.net/ly_twt/article/details/104879887

 

安装nvidia-docker2

（1）如果以前安装过docker，就需要先清理掉。

docker volume ls -q -f driver=nvidia-docker | xargs -r -I{} -n1 docker ps -q -a -f volume={} | xargs -r docker rm -f
sudo apt-get purge -y nvidia-docker

（2）添加包存储库

# 获取GPG密钥并添加密钥
curl -s -L https://nvidia.github.io/nvidia-docker/gpgkey | sudo apt-key add -

distribution=$(. /etc/os-release;echo $ID$VERSION_ID)

*******************************************************************************************

注：有的人执行该命令会出现以下错误：

Unsupported distribution!
Check https://nvidia.github.io/nvidia-docker

这是因为，distribution=$(. /etc/os-release;echo $ID$VERSION_ID)会获取当前系统的release版本，然后根据构建的地址从nvidia的仓库中获取源，但是nvidia仓库中没有https://nvidia.github.io/nvidia-docker/ubuntu16.10/nvidia-docker.list

解决办法：使用以下命令

# 定义变量distribution的值为ubuntu16.04
distribution="ubuntu16.04"

********************************************************************************************

然后：

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

（3）正式安装

sudo apt-get install -y nvidia-docker2

（4）

# 重新加载docker守护进程配置
sudo pkill -SIGHUP dockerd

（5）

# 测试
sudo docker run --runtime=nvidia --rm nvidia/cuda nvidia-smi

 

>>二更

1、docker加速

为了避免在拉取镜像时因网速问题而失败，执行以下内容以加速：

sudo mkdir -p /etc/docker

sudo tee /etc/docker/daemon.json <<-'EOF'
# 在终端输入以上命令后回车
>{
>"registry-mirrors": ["https://oyukeh0j.mirror.aliyuncs.com"]
>}
>EOF