确认NVIDIA驱动是否正确安装,并启用GPU;
确认已正确卸载以前的nvidia-docker,防止版本冲突;
安装nvidia-docker2,并重新启动Docker服务;
确认nvidia运行时是否正确安装,可以使用以下命令来检查运行时:
$ docker run --rm nvidia/cuda nvidia-smi
如果成功安装,则应该输出GPU信息;
如果还是出现“未知的运行时nvidia”的错误,则可以尝试手动注册nvidia运行时:
$ sudo docker run --rm --privileged
nvidia/cuda:9.0-base
nvidia-smi -L
检查注册表中是否存在“nvidia”运行时,如若不存在,则手动注册:
$ sudo docker run --rm --privileged
--runtime=nvidia
nvidia/cuda:9.0-base
nvidia-smi -L
该命令将在Docker的默认配置文件中添加“-gpus all”选项,以确保在使用GPU时正确使用nvidia运行时。