CentOS 7 安装 nvidia-docker2

nvidia-docker2 依赖 docker-ce安装前请确认已经安装好docker-ce, 可以参考 Centos 7 安装Docker

• 设置仓库

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

• 更新仓库中的key

  	DIST=$(sed -n 's/releasever=//p' /etc/yum.conf)
  	DIST=${DIST:-$(. /etc/os-release; echo $VERSION_ID)}
  	sudo yum makecache
  

• 安装nvidia-docker2

  	yum install nvidia-docker2
  

• 重新载入docker daemon.json文件，并重启docker

  	systemctl daemon-reload 
  	systemctl restart docker
  

• 运行如下命令进行测试

  	docker run --runtime=nvidia --rm nvidia/cuda nvidia-smi
  

• 首次运行需要下载nvidia/cuda组件，最后显示如下结果标识成功