centos 安装支持gpu的docker

要在 CentOS 上安装支持 GPU 的 Docker，你需要完成以下几个步骤：

1. **更新系统和安装必要的库**:

   sudo yum update -y
   sudo yum install epel-release -y  # 如果没有安装 EPEL（额外的第三方软件包仓库）
   sudo yum install docker-ce docker-ce-cli containerd.io -y

2. **安装 NVIDIA 驱动和工具**:
对于 NVIDIA GPU，首先安装 nvidia-docker 驱动。确保已经安装了 NVIDIA 显卡，并根据你的GPU型号和版本安装对应版本的驱动。例如：

   sudo yum install nvidia-container-toolkit -y
   sudo systemctl restart docker

3. **启用 NVIDIA 驱动对 Docker 的支持**:

   sudo nvidia-docker run --rm nvidia/cuda:latest nvidia-smi  # 测试GPU是否被识别
   echo 'nvidia.runtime = true' >> /etc/docker/daemon.json
   sudo systemctl restart docker

4. **验证安装**:
使用 `nvidia-docker run` 命令运行一个 GPU 兼容的镜像，如 `tensorflow/tensorflow:latest-gpu`，来确认一切正常。