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))
结果当然是不行,最后我查看这份代码的环境变量配置
我哭了,居然是环境变量最后多了一个空格。。。。。。。。。。。。。。。
然后就可以运行了