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

骑猪看日落 2022-02-26 01:04 342阅读 0赞
  1. 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

20190328162223985.png

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

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

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

watermark_type_ZmFuZ3poZW5naGVpdGk_shadow_10_text_aHR0cHM6Ly9ibG9nLmNzZG4ubmV0L2xydDM2Ng_size_16_color_FFFFFF_t_70

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

然后就可以运行了

watermark_type_ZmFuZ3poZW5naGVpdGk_shadow_10_text_aHR0cHM6Ly9ibG9nLmNzZG4ubmV0L2xydDM2Ng_size_16_color_FFFFFF_t_70 1

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

发表评论

表情:
评论列表 (有 0 条评论,342人围观)

还没有评论,来说两句吧...

相关阅读