nvidia镜像拉下来了,但是运行不了。
执行以下命令
sudo docker run --rm --runtime=nvidia --gpus all ubuntu nvidia-smi
报错
Failed to initialize NVML: Unknown Error
参考
https://stackoverflow.com/questions/72932940/failed-to-initialize-nvml-unknown-error-in-docker-after-few-hours
执行以下命令:
sudo vim /etc/nvidia-container-runtime/config.toml
设置 no-cgroups = false
,然后保存
sudo systemctl restart docker
sudo docker run --rm --runtime=nvidia --gpus all ubuntu nvidia-smi
成功显示: