Ubuntu18.04安装Nvidia-Docker2

为什么:安装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 

猜你喜欢

转载自blog.csdn.net/weixin_43332715/article/details/124422461