1. 问题描述
Error response from daemon: could not select device driver ““ with capabilities: [[gpu]]
本机运行镜像容器报错:找不到GPU。通常问题在于执行程序需要GPU,但是镜像中是没有GPU的,只能调用本机的GPU。
2. 解决方法
- 更新apt库,加载nvidia-docker(注意我这里是ubuntu18.04版本的)
$ curl -s -L https://nvidia.github.io/nvidia-docker/gpgkey | sudo apt-key add -
$ curl -s -L https://nvidia.github.io/nvidia-docker/ubuntu18.04/nvidia-docker.list | sudo tee /etc/apt/sources.list.d/nvidia-docker.list
$ sudo apt-get update
- 安装
nvidia-docker2
软件包并重新加载docker守护进程配置
$ sudo apt-get install nvidia-docker2
$ sudo pkill -SIGHUP dockerd
$ sudo systemctl restart docker
这里多说一句:在运行容器的时候调用GPU直接使用--gpus all
即可
参考
[1] https://blog.csdn.net/weixin_39132520/article/details/118676177
[2] https://blog.csdn.net/BigData_Mining/article/details/99681168