ubuntu18.04 部署GPU环境 docker启动完整步骤

1、首先安装nvidia驱动，根据自己的gpu版本下载对应的驱动，这里使用的gpu是2080Ti
驱动使用NVIDIA-Linux-x86_64-460.39.run

sh NVIDIA-Linux-x86_64-460.39.run

中间会让你安装gcc和make两个包，按提示安装即可

apt install gcc
apt install make

检测nvidia驱动是否安装成功：

nvidia-smi

显示下图则证明安装成功：

2、安装docker

apt install docker.io

安装成功之后，启动docker并设置开机自动启动

systemctl start docker
systemctl enable docker

查看docker 版本号

docker -V

3、安装toolkit
配置docker19使用gpu

# Add the package repositories
$ 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-container-toolkit
$ sudo systemctl restart docker

检测docker调用宿主机gpu

docker run --gpus=all nvidia/cuda:9.0-base nvidia-smi

出现下图安装成功

安装步骤不难，前提是有一篇这样的安装教程，否则会步步维艰，摸索不易。。。。。