運行ai模型的時候提示 no GPU available
執行 nvidia-smi 提示 Failed to initialize NVML: Unknown Error
解決方案
一. 宿主機執行sudo docker info
,確保 runtime 字段中有 nvidia
字樣
Runtimes: nvidia runc io.containerd.runc.v2
Default Runtime: runc
注意默認的還是 runc
二. sudo docker run
添加指令
- 指定運行時:
--runtime=nvidia
2. 授予權限:
--privileged
- 完整命令
sudo docker run --runtime=nvidia --privileged -ti -v ${PWD}:/mnt your_container /bin/bash
ok