CentOS里构建可以使用gpu的docker容器

主要是前期配置，docker容器我下载的是rstudio官方构建的，容器里面已经预先安装好了识别GPU需要的组件，不需要额外安装。(所有操作均在root账号下进行)
首先是CentOS服务器的更新，见上一篇文章，CentOS服务器更新源 - 简书
之后因为我们docker是之前安装和设置过的，所以不需要额外操作，先把docker服务停止即可：

docker stop [container id]

之后安装识别GPU所需要的nvidia-container-toolkit，首先设置源：

distribution=$(. /etc/os-release; echo $ID$VERSION_ID) \
&& curl -s -L https://nvidia.github.io/libnvidia-container/$distribution/libnvidia-container.repo | \
tee /etc/yum.repos.d/nvidia-container-toolkit.repo

安装完成后执行：

nvidia-ctk runtime configure --runtime=docker

之后重启docker：

systemcty restart docker

之后即可启动一个可以使用GPU的容器了：

docker run --name rstudio_and_gpu -d \
  -p 8790:8787 \
  -p 3840:3838 \
  -e USER=wrj \
  -e PASSWORD={password} \
  -e USERID=1005 \
  -e GROUPID=1006 \
  -e ADD=shiny \
  -v /data:/data \
  -v /data1:/data1 \
  --gpus all \
  docker.xuanyuan.run/rocker/rstudio:4.5.1

容器启动成功后进入容器执行：nvidia-smi，可以显示出显卡信息，说明容器构建成功。