如何在centos上安装nvidia docker

当基于nvidia gpu开发的docker镜像在实际部署时，需要先安装nvidia docker。那么如何安装nvidia docker呢。下面将详细介绍下。

安装原生docker

yum -y install docker-io

下载nvidia docker安装包

我下载的是rpm文件，具体见截图

安装nvidia docker安装包

cd 到rpm包内，使用命令 rpm -Uvh *.rpm --nodeps --force 完成安装。

运行docker服务

// 运行docker
systemctl start docker
// 加入开机启动
systemctl enable docker
// 查看状态
systemctl status coker

运行nvidia docker服务

systemctl start nvidia-docker
systemctl enable nvidia-docker
systemctl status nvidia-docker

nvidia-docker的操作命令与dock

加载nvidia 镜像

docker load < cuda_asr_docker.tar

docker image ls

[root@gpu3 asr]# docker image ls
REPOSITORY TAG IMAGE ID CREATED SIZE
64528942ab51 11 months ago 10GB

docker tag

docker tag 64528942ab51 asr_cuda:v1.
[root@gpu3 asr]# docker tag 64528942ab51 asr_cuda:v1.0
[root@gpu3 asr]# docker image ls
REPOSITORY TAG IMAGE ID CREATED SIZE
asr_cuda v1.0 64528942ab51 11 months ago 10GB

docker run

docker run -it -p 8087:8080 --restart=always --runtime=nvidia -e NVIDIA_VISIBLE_DEVICES=0 asr_cuda:v1.0 /bin/bash 试试

通过参数-runtime=nvidia及-e NVIDIA_VISIBLE_DEVICES=0来指定nvidia及gpu显卡ID。如果有多块显卡，NVIDIA_VISIBLE_DEVICES=1；NVIDIA_VISIBLE_DEVICES=2；
…
进行递增。

总结

当基于nvidia gpu开发的docker镜像在实际部署时，需要先安装nvidia docker。安装nvidia docker前需要先安装原生docker。本次我用的rpm包进行安装的。 与yum安装有区别。