【Docker】安装nvidia-container-runtime

使用Docker在GPU服务器上运行时报错

docker: Error response from daemon: could not select device driver "" with capabilities: [[gpu]].

这���由于服务器缺少 docker 使用 GPU 的 nvidia 的工具包支持，要使 docker 容器能调用本地的 gpu

解决方法

由于服务器是centos7，所以我们只需要下载合适的yum源，安装nvidia-container-runtime即可

# 获取Linux版本并赋值给变量distributiondistribution=$(. /etc/os-release;echo $ID$VERSION_ID)# 指定nvidia-container-runtime源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# 安装nvidia-container-runtimesudo yum install nvidia-container-runtime# 重启dockersystemctl restart docker

安装成功截图