tensorflow-gpu 使用GPU时,跟CPU情况下效率不变问题
设置使用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 技术交流群。
绑定邮箱获取回复消息
由于您还没有绑定你的真实邮箱,如果其他用户或者作者回复了您的评论,将不能在第一时间通知您!
发布评论
评论(2)
那看你在什么数据上跑了,要是小数据,本来就不需要多少计算量,反而数据在CPU和GPU之间来回传输还会拖慢计算。小数据量完全没必要用GPU,发挥不出计算优势。