错误之 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!

Logo

腾讯云面向开发者汇聚海量精品云计算使用和开发经验,营造开放的云计算技术生态圈。

更多推荐