在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的状态
[root@ourui]#systemctl start nvidia-docker
之后,将docker 看成一个单独的系统,进行安装所需的软件和包即可。