Docker 使用GPU时报错

问题:

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

出现这个问题的原因在于没有安装NVIDIA-docker的驱动

 

CentOS7 直接执行以下代码即可安装Docker使用GPU的驱动

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-container-toolkit