本文共 2005 字，大约阅读时间需要 6 分钟。

���CentOS���������������Docker���NVIDIA GPU������

������Docker���NVIDIA GPU���������������������������GPU���������������������������������������������������������������������������

������������

������������������������������������������������������

yum install -y yum-utils device-mapper-persistent-data lvm2

������������������������������������������������������������������������������������������������������������������

yum-config-manager --add-repo https://mirrors.aliyun.com/docker-ce/linux/centos/docker-ce.repo
yum makecache fast

������Docker

���������������Docker������������������

yum install -y docker-ce-19.03.12 docker-ce-cli-19.03.12 containerd.io

������Docker������������������������������������������

systemctl start docker
systemctl enable docker

������NVIDIA GPU������

���������GPU���������������������������NVIDIA������������������������������������������������

���������������������������������

distribution=$(. /etc/os-release; echo $ID$VERSION_ID)

������NVIDIA���������������

curl -s -L https://nvidia.github.io/nvidia-docker/$distribution/nvidia-docker.repo | sudo tee /etc/yum.repos.d/nvidia-docker.repo

������������NVIDIA������������������

yum install nvidia-container-runtime

������������

���������������������������������������NVIDIA ['-SMI������������GPU���������

docker run -d --gpus all --rm nvidia/cuda nvidia-smi

������Docker���������19������������������������������������������������

docker run --runtime=nvidia --rm nvidia/cuda:9.0-cudnn7-runtime-ubuntu16.04 nvidia-smi

������������������������������

lspci | grep -i vga
nvidia-smi

������CUDA���������

nvcc -V

������������������

������������������������������������������������������Docker���NVIDIA���������������������������������������������������������������

���������������������������������������

lspci | grep -i vga

������������������������������NVIDIA������������������NVIDIA���������������������������������������NVIDIA���������������

������������������������������������������������NVIDIA������������������������