为老版本Docker安装nvidia-docker

半兽人 发表于: 2019-08-20   最后更新时间: 2019-08-20 21:10:30  
{{totalSubscript}} 订阅, 4,640 游览

先决条件

运行nvidia-docker 2.0的先决条件列表如下所述。

  • GNU/Linux x86_64 内核版本 > 3.10
  • Docker >= 1.12
  • NVIDIA GPU with Architecture > Fermi (2.1)
  • NVIDIA drivers ~= 361.93 (旧版本未经测试)

你的驱动程序版本可能会限制你的CUDA功能(请参阅CUDA要求

安装2.0版

首先,确保你安装了NVIDIA驱动和Docker支持的版本。

如果您有自定义的/etc/docker/daemon.json,则nvidia-docker2包可能会覆盖它。

安装时,你必须固定nvidia-docker2nvidia-container-runtime的版本,例如:

sudo apt-get install -y nvidia-docker2=2.0.1+docker1.12.6-1 nvidia-container-runtime=1.1.0+docker1.12.6-1

使用 apt-cache madison nvidia-docker2 nvidia-container-runtimeyum search --showduplicates nvidia-docker2 nvidia-container-runtime 列出所有可用的版本。

实战操作如下:

## 我需要18.03版本的,所以我先查看我可用的版本
yum search --showduplicates nvidia-docker2 nvidia-container-runtime list|grep 18.03

安装

## 添加aliyun镜像库
yum-config-manager —add-repo http://mirrors.aliyun.com/docker-ce/linux/centos/docker-ce.repo
yum install -y nvidia-docker2-2.0.3-1.docker18.03.1.ce.noarch
yum install -y nvidia-container-runtime-2.0.0-1.docker18.03.1.x86_64

其他版本和体系结构

按照此处的说明分发安装存储库

测试

nvidia-docker运行时向Docker守护程序注册一个新的容器,使用docker run时必须选择nvidia

docker run --runtime=nvidia --rm nvidia/cuda nvidia-smi
更新于 2019-08-20
在线,5小时前登录

查看docker更多相关的文章或提一个关于docker的问题,也可以与我们一起分享文章