Ubuntu安装NVIDIA-Docker详细教程(离线&&在线)
docker支持nvidia
·
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
,显示信息则代表成功
更多推荐
已为社区贡献2条内容
所有评论(0)