2019-03-28 16:18:25.717636: E tensorflow/stream_executor/cuda/cuda_driver.cc:300] failed call to cuInit: CUDA_ERROR_INVALID_DEVICE: invalid device ordinal
运行代码时遇到这个报错,苦苦搜索,然而网上的答案我对照过都没有问题,这是服务器GPU状态,可以看到我这台服务器只有一块gpu。当前已经有人在运行了,所以gpu设备号只能是0

然后又随便试了试网上的办法
# 方法1
import os
os.environ['CUDA_VISIBLE_DEVICES'] = "0"
# 方法2,在创建session时使用这句代码:
sess = tf.Session(config=tf.ConfigProto(log_device_placement=True))
结果当然是不行,最后我查看这份代码的环境变量配置

我哭了,居然是环境变量最后多了一个空格。。。。。。。。。。。。。。。
然后就可以运行了


在运行代码时遇到CUDA_ERROR_INVALID_DEVICE错误,检查发现服务器GPU已分配给其他任务,设备号应为0。排查过程包括确认设备号指定正确及环境变量设置,最终发现是环境变量配置中多了一个空格导致问题,修正后GPU正常工作。
最低0.47元/天 解锁文章

5162

被折叠的 条评论
为什么被折叠?



