返回到文章

采纳

编辑于

为老版本Docker安装nvidia-docker

docker k8s kubernetes GPU
docker


先决条件

运行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