运行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