docker启动使用gpu的容器could not select with capabilities: [[gpu]].

本文地址:http://www.04007.cn/article/1116.html

    docker启动使用gpu的容器提示报错：docker: Error response from daemon: could not select device driver "" with capabilities: [[gpu]].本文地址:http://www.04007.cn/article/1116.html,未经许可,不得转载.

    我这里的服务器系统是Ubuntu18.04，问题是出在未安装nvidia-container-toolkit或nvidia-container-runtime(它包含nvidia-container-toolkit)。安装nvidia-container-toolkit需要添加nvidia-docker源，操作如下：本文地址:http://www.04007.cn/article/1116.html,未经许可,不得转载.

root@testMC:~/# distribution=$(. /etc/os-release;echo $ID$VERSION_ID)
root@testMC:~/# curl -s -L https://nvidia.github.io/nvidia-docker/gpgkey | sudo apt-key add
OK
root@testMC:~/# curl -s -L https://nvidia.github.io/nvidia-docker/$distribution/nvidia-docker.list | sudo tee /etc/apt/sources.list.d/nvidia-docker.list 
root@testMC:~/# apt-get update
root@testMC:~/# apt-get install nvidia-container-toolkit
#gpus选项提示
root@testMC:~/# $ docker run --help | grep -i gpus
      --gpus gpu-request               GPU devices to add to the container ('all' to pass all GPUs)
#安装好后可通过如下命令(两个之一都可以)在官方CUDA镜像上测试nvidia-smi
root@testMC:~/# docker run --runtime=nvidia --rm nvidia/cuda nvidia-smi
root@testMC:~/# docker run --gpus all --rm nvidia/cuda nvidia-smi
Status: Downloaded newer image for nvidia/cuda:latest

    安装好后通过上面任一命令在官方CUDA镜像上测试nvidia-smi，镜像启动成功并能正常显示GPU信息。可以指定使用几块GPU或指定使用哪块GPU进行工作。
#启动支持两块GPU的容器
$ sudo docker run –gpus 2 nvidia/cuda nvidia-smi
#指定某块GPU运行容器
$ sudo docker run –gpus device=0 nvidia/cuda nvidia-smi本文地址:http://www.04007.cn/article/1116.html,未经许可,不得转载.

本文地址:http://www.04007.cn/article/1116.html 未经许可,不得转载. 手机访问本页请扫描右下方二维码.

手机扫码直接打开本页面