nvidia镜像拉下来了,但是运行不了。
执行以下命令
sudo docker run --rm --runtime=nvidia --gpus all ubuntu nvidia-smi
报错
Failed to initialize NVML: Unknown Error
执行以下命令:
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
成功显示: