Ubuntu安装NVIDIA-Docker详细教程(离线&&在线)

在这里插入图片描述
B站|公众号:啥都会一点的研究生

Docker安装

虽然在前一贴详细说了安装步骤,Ubuntu安装、卸载Docker,常用镜像、容器操作命令,但还是将其总结写在此处

sudo apt-get remove docker docker-engine docker.io ###清除系统原有docker 如果提示找不到不用理会 

sudo apt-get update ###更新apt-get源 

sudo apt-get install apt-transport-https ca-certificates curl software-properties-common ###安装docker的依赖 

curl -fsSL https://download.docker.com/linux/ubuntu/gpg | sudo apt-key add -

sudo apt-key fingerprint 0EBFCD88

sudo add-apt-repository “deb [arch=amd64] https://download.docker.com/linux/ubuntu \ 
$(lsb_release -cs) \ 
stable”
 sudo apt-get update
 sudo apt-get install docker-ce docker-ce-cli containerd.io
sudo docker run hello-world #运行成功则表示安装成功

在线安装NVIDIA-Docker

https://github.com/NVIDIA/nvidia-docker/

需要代理才能安装。

首先确认已经安装好nvidia的驱动,可以用nvidia-smi命令查看

curl https://get.docker.com | sh
sudo systemctl start docker && sudo systemctl enable docker

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 update
sudo apt-get install -y nvidia-docker2
sudo systemctl restart docker

(推介)离线安装NVIDIA-Docker

已将离线文件打包放入云盘,为防止链接频繁挂掉,所以将其链接放在公众号【啥都会一点的研究生】,方便立即更新,回复【docker】即可获取
下载完毕后依次执行以下命令

sudo dpkg -i libnvidia-container1_1.2.0-1_amd64.deb
sudo dpkg -i libnvidia-container-tools_1.2.0-1_amd64.deb
sudo dpkg -i nvidia-container-toolkit_1.2.1-1_amd64.deb
sudo dpkg -i nvidia-container-runtime_3.3.0-1_amd64.deb
sudo dpkg -i nvidia-docker2_2.4.0-1_all.deb
sudo systemctl restart docker

注意

安装完nvidia-docker后,启动容器需要加上--gpus all
sudo docker run --rm --gpus all nvidia/cuda:11.0-base nvidia-smi
进入容器后输入nvidia-smi,显示信息则代表成功

猜你喜欢

转载自blog.csdn.net/zzh516451964zzh/article/details/128992983