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)
(2)重启:sudo service docker start
(3)在运行docker run -it中,将docker更改为nvidia-docker,即可运行。
(4)验证安装成功:使用nvidia-docker run -it 进入docker 环境中,
$ Nvidia-smi 即可
查看nvidia-docker的状态
![](https://img.haomeiwen.com/i12698522/a98ac0fbc06a4c16.png)
[root@ourui]#systemctl start nvidia-docker
之后,将docker 看成一个单独的系统,进行安装所需的软件和包即可。