Nvidia docker 是nvidia显卡在docker基础上进行封装得到的docker工具，需要电脑中安装Nvidia显卡驱动与docker，配置好Nvidia docker后docker可以使用GPU。本文记录Nvidia docker的安装与使用方法。

环境 {#环境}

• Linux 16.04 64位操作系统
• 显卡驱动 450.80.02
• CUDA 版本 11.0
• docker 安装版本 19.03.4

安装过程 {#安装过程}

• 加入包仓库

好像不好使了，可以用备份库

• 安装nvidia-docker2 载入docker 配置

• 如果安装失败可以尝试加入参数 --fix-missing

• 重启 docker

• 下载官方测试镜像

其中 DIST 为 ubuntu20.04, ubuntu18.04, centos7 其中之一。

nvidia/cuda:11.0-base 镜像已经被移除

• 例如

此时会显示出显卡信息，说明nvidia docker成功创建并在内部正确执行了 nvidia-smi 命令。

• 查看nvidia-docker安装情况

出现类似信息说明安装成功

使用nvidia docker镜像 {#使用nvidia-docker镜像}

• 之前不是同一个驱动版本下创建的镜像在新驱动下的nvidia docker中可能找不到nvidia-smi命令
• 为了在新的docker下使用gpu，我迂回地使用测试镜像作为初始镜像

其中 DIST 为 ubuntu20.04, ubuntu18.04, centos7 其中之一。

这样便将测试镜像创建出了可以随时访问修改的容器，在该容器基础上保存镜像即可。

参考资料 {#参考资料}

• https://www.cnblogs.com/llfctt/p/10642100.html

• https://www.cnblogs.com/wuchangsoft/p/9767074.html

• https://github.com/NVIDIA/nvidia-docker/issues/1735

• https://zhuanlan.zhihu.com/p/361934132

文章链接：
https://www.zywvvd.com/notes/tools/docker/nvidia-docker-install/