CentOS上Docker支持GPUs

目录

前言

一、先决条件

二、用户管理

三、GPU访问配置

四、资源配置

GPU

CPU

内存

IP

磁盘

总结

---

前言

Docker 19.03后增加了gpus配置，无需安装nvidia-docker，即可访问并配置GPU。

---

一、先决条件

具体操作见：Docker访问GPU的先决条件。

 

二、用户管理

正常情况下，只有root权限才能访问Docker。

为了系统的安全性和资源的统一管理，利用root权限开放Docker权限，并分配到不同用户。

具体操作指令如下：

 

# 增加用户组
groupadd docker

# 增加docker用户
gpasswd -a [user]

# 查看用户组
cat /etc/group |grep docker

# 重启docker
systemctl start docker && systemctl enable docker

# 进入用户
su [user]

# 本用户下查看docker信息
docker info

1、root权限和用户权限下的docker信息一致，对比如下：

 

三、GPU访问配置

不同Linux系统的GPU访问配置教程：Nvidia显卡配置

CentOS是RHEL系列，因此配置如下：

# 编辑文件
vim centos_nvidia-container-runtime
"""
distribution=$(. /etc/os-release;echo $ID$VERSION_ID)
curl -s -L https://nvidia.github.io/nvidia-container-runtime/$distribution/nvidia-container-runtime.repo | \
  sudo tee /etc/yum.repos.d/nvidia-container-runtime.repo
"""

# 修改权限
chmood +x centos_nvidia-container-runtime

# 执行文件
sh centos_nvidia-container-runtime

# 查看生成的配置文件
ls /etc/yum.repos.d/nvidia-container-runtime.repo
cat /etc/yum.repos.d/nvidia-container-runtime.repo

# 重启docker
systemctl start docker && systemctl enable docker

# 测试GPU介质
docker run -d  -it --name="centos"  --gpus all nvidia/cuda:9.1-cudnn7-runtime-centos7 /bin/bash

# 进入docker
dockers attach centos

# 查看显卡信息
nvidia-smi

1、配置文件centos_nvidia-container-runtime内容如下：

2、生成配置文件nvidia-container-runtime.repo内容如下：

3、GPU介质测试结果

4、服务器和docker的GPU信息比对

 

四、资源配置

• GPU

参数：--gpus

• CPU

参数：--cpuset-cpus

• 内存

参数：-m

• IP

参数：-p

• 磁盘

参数：-v

 

---

总结

Docker下支持GPU的多用户资源管理与配置。