用docker创建任务,指定使用gpu的时候,报错。国外网友给出的答案解决了问题,记录一下。
-
distribution=$(. /etc/os-release;echo I D ID IDVERSION_ID) && curl -s -L https://nvidia.github.io/libnvidia-container/gpgkey | sudo apt-key add - && curl -s -L https://nvidia.github.io/libnvidia-container/$distribution/libnvidia-container.list | sudo tee /etc/apt/sources.list.d/nvidia-container-toolkit.list
-
sudo apt-get update
-
sudo apt-get install -y nvidia-docker2
-
sudo systemctl restart docker
文章地址:
https://askubuntu.com/questions/1400476/docker-error-response-from-daemon-could-not-select-device-driver-with-capab