更新GPU驱动程序后,我无法在Docker上运行GPU,当我在主机环境(Centos)中运行nvidia-smi
时,GPU被识别。docker run --gpus all -it -v $(pwd):/home/workspace test /bin/bash
个
ERRO[0000] error waiting for container: context canceled
docker: Error response from daemon: could not select device driver "" with capabilities: [[gpu]].
从我所研究的,似乎更新GPU驱动程序删除设置,所以我参考NVIDIA container toolkit网站,并再次执行安装程序,但上述错误仍然存在。我也重新启动系统只是为了确定,但这并没有解决问题。我该怎么办?docker run -it --gpus all nvidia/cuda:11.4.0-base-ubuntu20.04 nvidia-smi
这也会产生上述错误。
1条答案
按热度按时间qhhrdooz1#
假设您遵循油井安装程序:
现在,从教程:如何连接NVIDIA GPU与Docker https://www.howtogeek.com/devops/how-to-use-an-nvidia-gpu-with-docker-containers/
使用nvidia基本映像而不是您的映像
test
如果运行不好,您的映像
test
需要进一步调试