ubuntu nvidia-docker 一键安装 脚本

ubuntu  nvidia-docker 一键安装 脚本 

#!/bin/bash
#libc6=2.23-0ubuntu10 		libc6-dbg	libc6-dev
#libcap2=1:2.24-12	libcap2-bin
#libnvidia-container1=1.0.0~rc.2-1
#libseccomp2=2.3.1-2.1ubuntu2~16.04.1
#libgcc1=1:6.0.1-0ubuntu1
#libnvidia-container-tools=1.0.0~rc.2-1


#### install docker-ce #####
apt-get update
apt-get install -y curl
curl -fsSL http://mirrors.aliyun.com/docker-ce/linux/ubuntu/gpg | sudo apt-key add -
add-apt-repository "deb [arch=amd64] http://mirrors.aliyun.com/docker-ce/linux/ubuntu $(lsb_release -cs) stable"
apt-get -y update
apt-get install -y docker-ce=18.06.0~ce~3-0~ubuntu
groupadd docker
usermod -aG docker $USER


#### install nvidia-docker #####
docker volume ls -q -f driver=nvidia-docker | xargs -r -I{} -n1 docker ps -q -a -f volume={} | xargs -r docker rm -f
curl -s -L https://nvidia.github.io/nvidia-docker/gpgkey | \
   apt-key add -
distribution=$(. /etc/os-release;echo $ID$VERSION_ID)
curl -s -L https://nvidia.github.io/nvidia-docker/$distribution/nvidia-docker.list | \
   tee /etc/apt/sources.list.d/nvidia-docker.list
apt-get update

apt-get install -y nvidia-container-runtime=2.0.0+docker18.06.0-1
apt-get install -y nvidia-docker2=2.0.3+docker18.06.0-1


#### add docker hub mirror ####
cat << EOF > test.txt
{
    "registry-mirrors": ["https://registry.docker-cn.com"],
    "default-runtime": "nvidia",
    "runtimes": {
        "nvidia": {
            "path": "/usr/bin/nvidia-container-runtime",
            "runtimeArgs": []
        }
    }
}
EOF
service docker restart

运行脚本即可