CentOS7 使用 Nvidia-Docker

Docker要使用GPU之前是一件麻烦事

但是现在自从Docker 19.03版本之后就变的方便了很多

 

环境：CentOS 7.9 x64

Docker CE 20.10

 

直接上干货

Docker安装就不多说了，网上一堆教程

显卡需要先安装驱动程序（安装了驱动后无需单独安装CUDA）

如果不会安装显卡驱动的可以参考我另外一篇文章 https://blog.csdn.net/qq_35422558/article/details/115747760?spm=1001.2014.3001.5501

 

要让Docker使用GPU就需要用到nvidia-docker 这个插件

官方地址

https://nvidia.github.io/nvidia-docker/

 

紧接着还需要将nvidia-docker的源给放在本地，以下是官方给出的地址

distribution=$(. /etc/os-release;echo $ID$VERSION_ID)
curl -s -L https://nvidia.github.io/nvidia-docker/$distribution/nvidia-docker.repo | \
  sudo tee /etc/yum.repos.d/nvidia-docker.repo

 

连不上就更改Host(失效了给我留言,我会及时更新)

185.199.109.153 nvidia.github.io

 

接着就开始安装nvidia-docker

yum install nvidia-docker2

导入密钥，一直输入y 然后回车即可

如果以上方法还是不能用，则需要用离线安装的方式去安装，RPM包依赖如下，用另外的能连“外网”的的电脑去下载后传给Linux就行

安装完毕了输入

nvidia-docker -v

出现版本号就成功了

最后重启docker即可

sudo systemctl restart docker