关于使用服务器gpu的问题,cuda_driver.cc:300] failed call to cuInit: CUDA_ERROR_INVALID_DEVICE: invalid device o

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

关于使用服务器gpu的问题,cuda_driver.cc:300] failed call to cuInit: CUDA_ERROR_INVALID_DEVICE: invalid device o_第1张图片

然后又随便试了试网上的办法

# 方法1
import os
os.environ['CUDA_VISIBLE_DEVICES'] = "0"
# 方法2,在创建session时使用这句代码:
sess = tf.Session(config=tf.ConfigProto(log_device_placement=True))

结果当然是不行,最后我查看这份代码的环境变量配置

关于使用服务器gpu的问题,cuda_driver.cc:300] failed call to cuInit: CUDA_ERROR_INVALID_DEVICE: invalid device o_第2张图片

我哭了,居然是环境变量最后多了一个空格。。。。。。。。。。。。。。。

然后就可以运行了

关于使用服务器gpu的问题,cuda_driver.cc:300] failed call to cuInit: CUDA_ERROR_INVALID_DEVICE: invalid device o_第3张图片

总结:gpu不能使用,先看设备号有没有指定对,然后看环境变量有没有配好

你可能感兴趣的:(关于使用服务器gpu的问题,cuda_driver.cc:300] failed call to cuInit: CUDA_ERROR_INVALID_DEVICE: invalid device o)