Centos7安装docker支持NVIDIA GPU

Centos安装docker

步骤如下

yum update -y
yum install -y yum-utils device-mapper-persistent-data lvm2
yum-config-manager --add-repo https://download.docker.com/linux/centos/docker-ce.repo
yum install -y docker-ce docker-ce-cli containerd.io

# GPU相关
distribution=$(. /etc/os-release;echo $ID$VERSION_ID)
curl -s -L https://nvidia.github.io/nvidia-docker/$distribution/nvidia-docker.repo | tee /etc/yum.repos.d/nvidia-docker.repo

# 安装nvidia-docker2
yum install -y nvidia-docker2

# 重启Docker服务
systemctl restart docker
systemctl enable docker

修改docker Root dir

由于默认的docker Root dir一般在/，空间不够。需要修改成：

mkdir /home/docker-data
systemctl stop docker
vi /etc/docker/daemon.json
#内容如下
{
  "data-root": "/home/docker-data"
}
# 查看是否修改成功
docker info | grep "Docker Root Dir"

创建能用GPU的容器

–gpus=all: 使用host上所有的GPU，可以通过id指定单独的某个GPU
–net host：使用host的网络
-d: 参数表示 “detached mode”，即在后台运行容器并打印容器 ID。
-v PATH:PATH: 挂载host目录

docker run -d --net host --gpus=all -v /mnt/disk0/:/mnt/disk0 -it IMAGE_PATH:VERSION /bin/bash