docker: Error response from daemon: could not select device driver ““ with capabilities: [[gpu]]问题解决

Posted 2023-03-01 一个菜鸟的奋斗

docker在19.03版本之后，可以不用安装nvidia-docker，就能获得GPU的计算支持。

但是，直接在docker中使用GPU设备，出现报错如下：

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

这个问题可以通过安装nvidia-container-toolkit来解决。

1、添加nvidia-docker的源

curl -s -L https://nvidia.github.io/nvidia-docker/gpgkey | \\
  sudo apt-key add -
distribution=$(. /etc/os-release;echo $ID$VERSION_ID)
curl -s -L https://nvidia.github.io/nvidia-docker/$distribution/nvidia-docker.list | \\
  sudo tee /etc/apt/sources.list.d/nvidia-docker.list
sudo apt-get update

如果不添加上面的源，可能会报如下错误：

E: Unable to locate package nvidia-container-toolkit

2、安装nvidia-container-toolkit

sudo apt-get install -y nvidia-container-toolkit

3、重启docker

sudo systemctl restart docker

问题解决。