weixin_38946936
weixin_38946936
采纳率83.3%
2018-11-08 07:30 阅读 15.7k

tensorflow-gpu为何无法调用GPU进行运算?

5

如题,本人是小白级别的爱好者,使用的是联想台式机,win10系统,有一块GeForce GT730的独立显卡,想尝试安装tensorflow-gpu 进行加速。
在参考官网方法后,升级了显卡驱动,安装了CUDA9.0 及配套的cudnn7
并添加了环境变量。然后pip 安装tensorflow-gpu
安装成功后,import tensorflow as tf 不报错,但是运行如下代码时,始终显示GPU使用率为0

import tensorflow as tf
with tf.device('/cpu:0'):
    a=tf.constant([1.0,2.0,3.0,4.0,5.0,6.0],shape=[2,3],name='a')
    b=tf.constant([1.0,2.0,3.0,4.0,5.0,6.0],shape=[3,2],name='b')
    c=tf.matmul(a, b)
    sess=tf.Session(config=tf.ConfigProto(allow_soft_placement=False, log_device_placement=True))
    print(sess.run(c))

试用如下代码检查是否有GPU可以被使用:

import os
from tensorflow.python.client import device_lib
os.environ["TF_CPP_MIN_LOG_LEVEL"] = "99"

if __name__ == "__main__":
    print(device_lib.list_local_devices())

显示 只有一个CPU可以被调用

[name: "/device:CPU:0"
device_type: "CPU"
memory_limit: 268435456
locality {
}
incarnation: 15723487639721858299
]

那么问题来了。。既然已经成功安装了tensorflow-gpu,为什么仍然无法调用gpu进行计算呢。。

而且,装好tensorflow-gpu之后,双击程序会闪退,但是从IDLE中run是可以运行的,也不报错,但就是不分配给GPU运算。

查看了一下cuda,显示GPU not supported
图片说明
这就愈发郁闷了。。。。。。

深知肯定是自己还有什么地方没设置好,但是网上也找不到对应的教程了,只好在此想各位大了!!!!
万望赐教!!!!
感激不尽!!!!

  • 点赞
  • 写回答
  • 关注问题
  • 收藏
  • 复制链接分享

5条回答 默认 最新

  • 已采纳
    caozhy 从今以后生命中的每一秒都属于我爱的人 2018-11-08 10:08

    https://ask.csdn.net/questions/710166 回答你了,如果满意,请点我回答左上角的箭头和采纳。谢谢

    点赞 评论 复制链接分享
  • faruba faruba 2019-06-11 14:37

    with tf.device('/cpu:0'):
    难道不是因为这句话么?

    点赞 6 评论 复制链接分享
  • qq_32734095 中国好码农 2019-03-08 16:47

    我的也是,出现跟你一样的问题,运行发现:Ignoring visible gpu device (device: 0, name: GeForce 820M, pci bus id: 0000:08:00.0, compute capability: 2.1) with Cuda compute capability 2.1. The minimum required Cuda capability is 3.0.意思大概是我的GPU运算能力只有2.1,而Tensorflow对GPU的运算能力要求至少为3.0,上网搜一下对照表自己看看是不是这个原因。

    点赞 2 评论 复制链接分享
  • xiaomad xiaomad 2018-11-08 07:54

    cudnn7? 为啥不装9?

    点赞 1 评论 复制链接分享
  • weixin_47390487 deeplearndrz 2021-04-02 19:50
    点赞 评论 复制链接分享

相关推荐