深度学习

Linux下开发环境搭建

3. Nvidia-docker环境安装

官方文档:https://docs.nvidia.com/datacenter/cloud-native/container-toolkit/install-guide.html#docker

参考文档:https://zhuanlan.zhihu.com/p/64493662

https://blog.csdn.net/weixin_43002433/article/details/108888927?depth_1-

  • 按照官网步骤,看一下,我们的系统Ubuntu 16.01.7 LTS的,是符合要求的,下面开始安装
    在这里插入图片描述

  • 设置存储库和GPG密钥:

distribution=$(. /etc/os-release;echo $ID$VERSION_ID) \
   && curl -s -L https://nvidia.github.io/nvidia-docker/gpgkey | sudo apt-key add - \
   && curl -s -L https://nvidia.github.io/nvidia-docker/$distribution/nvidia-docker.list | sudo tee /etc/apt/sources.list.d/nvidia-docker.list

好像报错了:gpg: no valid OpenPGP data found.

  • 那就编辑hosts文件
sudo vi /etc/hosts
  • 然后将下面的内容写入hosts文件
# nvidia.github.io
185.199.108.153 nvidia.github.io
185.199.109.153 nvidia.github.io
185.199.110.153 nvidia.github.io
185.199.111.153 nvidia.github.io

在这里插入图片描述

  • 再执行设置存储库和GPG密钥
distribution=$(. /etc/os-release;echo $ID$VERSION_ID) \
   && curl -s -L https://nvidia.github.io/nvidia-docker/gpgkey | sudo apt-key add - \
   && curl -s -L https://nvidia.github.io/nvidia-docker/$distribution/nvidia-docker.list | sudo tee /etc/apt/sources.list.d/nvidia-docker.list

在这里插入图片描述

  • 安装软件包(和依赖项)
sudo apt-get install -y nvidia-docker2

在这里插入图片描述

  • 设置默认运行时后,重新启动Docker守护程序以完成安装
sudo systemctl restart docker

在这里插入图片描述

  • 此时,可以通过运行基本CUDA容器来测试有效的设置,显示图片中的表格安装成功。
sudo docker run --rm --gpus all nvidia/cuda:11.0-base nvidia-smi

在这里插入图片描述
在这里插入图片描述

  • 按照官网教程,需要添加NVIDIA运行时,但是在nvidia-docker2里,已经默认注册了,就不用管了,可以看一下。
    在这里插入图片描述
cat /etc/docker/daemon.json

在这里插入图片描述

创作不易,喜欢的话加个关注点个赞,❤谢谢谢谢❤

更多推荐