Nvidia-docker安装

2018-06-21  本文已影响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 看成一个单独的系统,进行安装所需的软件和包即可。

上一篇 下一篇

猜你喜欢

热点阅读