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 未经许可,不得转载. 手机访问本页请扫描右下方二维码.
![]() |
![]() |
手机扫码直接打开本页面 |