CENTOS7 docker 中使用CUDA:

参考：https://docs.nvidia.com/datacenter/cloud-native/container-toolkit/install-guide.html#docker

1. 安装显卡驱动
   NVIDIA-Linux-x86_64-470.57.02.run
   安装过程中要求kernel 头文件 yum 直接安装即可。
   yum install kernel-devel-$(uname-r)kernel-headers-$(uname -r)

2. 安装docker
   卸载旧版本： sudo yum remove docker
   docker-client
   docker-client-latest
   docker-common
   docker-latest
   docker-latest-logrotate
   docker-logrotate
   docker-engine
   安装 sudo yum-config-manager
   –add-repo
   https://download.docker.com/linux/centos/docker-ce.repo

   sudo yum install docker-ce docker-ce-cli containerd.io

3. 安装 NVIDIA Container Toolkit
   distribution=$(. /etc/os-release;echo $ID$VERSION_ID)
   && curl -s -L https://nvidia.github.io/nvidia-docker/$distribution/nvidia-docker.repo | sudo tee /etc/yum.repos.d/nvidia-docker.repo

yum install -y nvidia-docker2
systemctl restart docker

docker run --rm --gpus all nvidia/cuda:11.0-base nvidia-smi
上面命令如果正常运行，就表示docker 安装成功。 下载一个nvidia 镜像，进行开发即可。