美文网首页
Nvidia-docker安装

Nvidia-docker安装

作者: 兜里有颗糖儿 | 来源:发表于2018-06-21 13:40 被阅读0次

在docker中若要调用gpu,则需要通过nvidia-docker 来将docker内外的gpu连接起来。


Nvidia-docker 安装

(1)sudo curl -s -L https://nvidia.github.io/nvidia-docker/centos7/nvidia-docker.repo| \

sudo tee/etc/yum.repos.d/nvidia-docker.repo

(安装的版本为nvidia-docker-1.0.1-1.x86_64.rpm)

yum install -y nvidia-docker

(2)重启:sudo service docker start

(3)在运行docker run -it中,将docker更改为nvidia-docker,即可运行。

(4)验证安装成功:使用nvidia-docker run -it 进入docker 环境中,

$ Nvidia-smi 即可


查看nvidia-docker的状态

[root@ourui]#systemctl start nvidia-docker

之后,将docker 看成一个单独的系统,进行安装所需的软件和包即可。

相关文章

网友评论

      本文标题:Nvidia-docker安装

      本文链接:https://www.haomeiwen.com/subject/xlvqyftx.html