为什么:安装Nvidia-Docker2的目的一般是为了让Docker容器中的程序在GPU上跑
如果你还没有在Ubuntu上安装Docker,请先去这里参考安装一下:
Ubuntu18.04下安装Docker【亲测可用】_思考实践的博客-CSDN博客
安装好后,继续安装Nvidia-Docker2
curl -s -L https://nvidia.github.io/nvidia-docker/gpgkey | sudo apt-key add -
distribution=$(. /etc/os-release;echo $ID$VERSION_ID)
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 update
安装nvidia-docker2并重新载入daemon.json
sudo apt-get install -y nvidia-docker2
重新加载docker守护程序配置
sudo pkill -SIGHUP dockerd
验证是否安装完毕
nvidia-docker --version
这篇的目的是为了通过Docker拉取Pytorch官方镜像通过实例化镜像,得到容器,在容器(沙盒)里面运行深度学习程序并且调用本地GPU,请在Docker专栏查找获取
参考资料
Ubuntu 安装Docker和Pythorch_Sakuya__的博客-CSDN博客
Ubuntu中,在docker中安装pytorch(gpu)+cuda-详细_管牛牛的博客-CSDN博客_docker 容器中安装cuda