nvidia-docker安装
Nvidia-Docker安装需要安装两个部分,Docker-CE和NVIDIA Container Toolkit。
1. 安装Dokcer-CE
Docker-CE on Ubuntu can be setup using Docker’s official convenience script:
官方的快速安装脚本,具体安装的版本应该是最新版,如果用此脚本安装Docker,以后还可以使用此脚本更新:
curl https://get.docker.com | sh
&& sudo systemctl --now enable docker
注意:如果你的机器已经有安装Docker,那么使用这个脚本可能会出错,要是你本机的Docker就是用这个脚本安装的,那么你可以再次用这个脚本进行更新。
我使用此脚本进行安装后:
需要执行sudo /lib/systemd/systemd-sysv-install enable docker
来激活docker,激活后查看Docker版本docker --version
2. 安装NVIDIA Container Toolkit
设置稳定版本的库及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
Optional:如果想要使用实验特性,需要加入experimental分支到库下:
curl -s -L https://nvidia.github.io/nvidia-container-runtime/experimental/$distribution/nvidia-container-runtime.list | sudo tee /etc/apt/sources.list.d/nvidia-container-runtime.list
更新好包列表之后,安装nvidia-docker2包及其依赖:
sudo apt-get update
sudo apt-get install -y nvidia-docker2
安装完毕…
上述默认运行时设置好后,重启Docker后台驻留程序:
sudo systemctl restart docker
现在可以通过运行base CUDA container来测试一个working setup文章来源:https://www.toymoban.com/news/detail-599087.html
sudo docker run --rm --gpus all nvidia/cuda:11.0-base nvidia-smi
控制台的输出如下:
此时这个实例就已经在Docker的镜像库中了:使用sudo docker images -a
查看已下载的镜像:
文章来源地址https://www.toymoban.com/news/detail-599087.html
到了这里,关于nvidia-docker安装教程的文章就介绍完了。如果您还想了解更多内容,请在右上角搜索TOY模板网以前的文章或继续浏览下面的相关文章,希望大家以后多多支持TOY模板网!