ubuntu18上docker支持GPU

一、系统环境

ununtu18.04+

如果docker已经安装并且docker --version命令显示版本大于等于19.03，则请忽略安装docker 

二、安装docker

sudo apt update 
#更新可用软件包列表 
sudo apt -y upgrade 
#更新所有软件包 
sudo apt install -y docker.io 
#安装 docker 
docker --version 
#确认 docker 版本大于等于 19.03

三、安装 Nvidia container toolkit(仅 GPU 需要)

Nvidia container toolkit 使用户能在 docker 容器内部访问宿主机上的 Nvidia GPU。 

1、系统要求

GNU/Linux x86_64 with kernel version > 3.10 
Docker >= 19.03 
NVIDIA GPU with Architecture > Fermi (or compute capability 2.1) 
NVIDIA drivers ~= 361.93 (untested on older versions) 
推荐 NVIDIA drive 435.21+ 
安装 NVIDIA drive 的方法： 
下载驱动： 
wget \ 
http://us.download.nvidia.com/XFree86/Linux-x86_64/435.21/NVIDIA-Linux-x86_64-435.21.ru n  
安装驱动： 
sudo sh NVIDIA-Linux-x86_64-435.21.run 
测试驱动: 
nvidia-smi 

2、安装步骤

distribution=$(. /etc/os-release;echo $ID$VERSION_ID) #得到 ubuntu 发型版的代号 

curl -s -L https://nvidia.github.io/nvidia-docker/gpgkey | \ 
sudo apt-key add - 
#增加 nvidia 的 apt key 
curl -s -L \ 
https://nvidia.github.io/nvidia-docker/$distribution/nvidia-docker.li st | \ 
sudo tee /etc/apt/sources.list.d/nvidia-docker.list 
#增加 nvidia 的 apt 源 

sudo apt-get update #更新可用软件包列表

sudo apt-get install -y nvidia-docker2 
#安装 nvidia docker container toolkit 
sudo systemctl restart docker 
#重启 docker 服务 
sudo docker run --rm --gpus all nvidia/cuda:11.0-base nvidia-smi #在 docker 容器内运行 nvidia-smi，测试安装是否成功 
#如果 nvidia-smi 能正确显示 GPU 状态，则安装成功