运行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
- 授予权限:
--privileged
如果不设置这个,在容器中执行 nvidia-smi 的时候会出现错误:
nvidia-smi
Failed to initialize NVML: Unknown Error
- 设置端口映射(宿主机端口映射到容器端口)【与此处问题无关,给ssh 作准备】
-p 2200:22
- 完整命令
sudo docker run -p 2200:22 --runtime=nvidia --privileged -ti -v ${PWD}:/mnt gyxthu17/cpm-2:1.2 /bin/bash
ok