侧边栏壁纸
博主头像
故黎

Talk is cheap , show me the code.

  • 累计撰写 27 篇文章
  • 累计创建 4 个标签
  • 累计收到 7 条评论
标签搜索

CentOS 7 安装 nvidia-docker2

故黎
2021-08-06 / 0 评论 / 2 点赞 / 884 阅读 / 571 字 / 正在检测是否收录...
温馨提示:
本文最后更新于 2021-12-30,若内容或图片失效,请留言反馈。部分素材来自网络,若不小心影响到您的利益,请联系我们删除。

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组件,最后显示如下结果标识成功
    image.png

0
广告 广告

评论区