Docker 使用GPU 错误: Error could not select device driver ““ with capabilities: [[gpu]] 解决方案 -2025 验证可用
【代码】Docker 使用GPU 错误: Error could not select device driver ““ with capabilities: [[gpu]] 解决方案 -2025 验证可用。
·
错误之 Error response from daemon: could not select device driver ““ with capabilities: [[gpu]]
关于配置docker使用gpu,其实只用装官方提供的toolkit即可,把github上的搬下来:
Ubuntu 16.04/18.04/22.04 , Debian Jessie/Stretch/Buster:
distribution=$(. /etc/os-release;echo $ID$VERSION_ID)
curl -s -L https://nvidia.github.io/nvidia-docker/gpgkey | sudo apt-key add -
curl -s -L https://nvidia.github.io/nvidia-docker/$distribution/nvidia-docker.list | sudo tee /etc/apt/sources.list.d/nvidia-docker.list
sudo apt-get update && sudo apt-get install -y nvidia-container-toolkit
sudo systemctl restart docker
验证:which nvidia-container-runtime ,是否出现:
/usr/bin/nvidia-container-runtime
运行:
docker run --rm -it --init \
--gpus=all \
--ipc=host \
--user="$(id -u):$(id -g)" \
--volume="$PWD:/app" \
anibali/pytorch python3 main.py
OK!
更多推荐

所有评论(0)