docker使用GPU报错“Unknown runtime specified nvidia“

问题1复现

• 系统：ubuntu18.04
• docker 版本： 20.10.7
  当我启动一个容器时，运行以下命令：

docker run -itd \
   --runtime=nvidia --gpus=all \
   -e NVIDIA_DRIVER_CAPABILITIES=compute,utility,video,graphics \
   image_name
   

报错：

docker: Error response from daemon: Unknown runtime specified nvidia.
See 'docker run --help'.

解决方案1

这是因为用户没有加入到docker组中，将自己的用户加入到docker的用户组。

sudo usermod -a -G docker $USER

问题2复现

docker: Error response from daemon: Unknown runtime specified nvidia.
See 'docker run --help'.

解决方案2

需要安装nvidia-docker2

sudo apt-get install -y nvidia-docker2

将docker重启

sudo systemctl daemon-reload
sudo systemctl restart docker