tensorflow-gpu 使用GPU时,跟CPU情况下效率不变问题

发布于 2022-09-06 22:04:34 字数 929 浏览 33 评论 0

设置使用GPU,但是计算速度没有什么变化,gpu利用率极低。
下面是部分代码:


from tensorflow.python.framework import ops

ops.reset_default_graph()

config = tf.ConfigProto() 
config.gpu_options.per_process_gpu_memory_fraction = 0.85
config.gpu_options.allow_growth = True
sess = tf.Session(config=config)

with tf.device('/GPU:0'):
    x = tf.placeholder(shape=[None, 20],dtype=tf.float32, name="x_m")
    y = tf.placeholder(shape=[None, 2], dtype=tf.float32, name="y_m")

    with tf.name_scope("cross_entropy"):
        y_ = train_model_gpu(x)
        cross_entropy = tf.reduce_mean(tf.nn.softmax_cross_entropy_with_logits(logits=y_, labels=y), name="cross_entropy_m")

图片描述

图片描述

如果你对这篇内容有疑问,欢迎到本站社区发帖提问 参与讨论,获取更多帮助,或者扫码二维码加入 Web 技术交流群。

扫码二维码加入Web技术交流群

发布评论

需要 登录 才能够评论, 你可以免费 注册 一个本站的账号。

评论(2

想念有你 2022-09-13 22:04:34
  1. 任务管理器默认显示3D、Copy这几项,但是在用cuda的时候,主要使用的是Compute和Copy,你要切换一下
  2. Copy只有2%左右,数据量不大,模型太简单也发挥不了GPU优势
你是暖光i 2022-09-13 22:04:34

那看你在什么数据上跑了,要是小数据,本来就不需要多少计算量,反而数据在CPU和GPU之间来回传输还会拖慢计算。小数据量完全没必要用GPU,发挥不出计算优势。

~没有更多了~
我们使用 Cookies 和其他技术来定制您的体验包括您的登录状态等。通过阅读我们的 隐私政策 了解更多相关信息。 单击 接受 或继续使用网站,即表示您同意使用 Cookies 和您的相关数据。
原文