GPU 调用问题:failed call to cuInit: CUDA_ERROR_NO_DEVICE: no CUDA-capable device is detected

今天直接拿别人的测试代码来跑的时候,出现了上述问题:

failed call to cuInit: CUDA_ERROR_NO_DEVICE: no CUDA-capable device is detected

字面意思是GPU不可用。

但是我nvdia-smi了一下,发现服务器几个gpu都是空的啊。不可能啊。而且我的运行有加前缀:

CUDA_VISIBLE_DEVICES=2 python test.py

思来想去,查来查去,还是CUDA_VISIBLE_DEVICES的问题。原来代码隐藏的某处,把这个数值写死了,写成了

CUDA_VISIBLE_DEVICES=-1意为使用cpu。。。

命令行传进去的值被里面覆盖了。好吧。改写后就正常了。

评论 4
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值