在线观看不卡亚洲电影_亚洲妓女99综合网_91青青青亚洲娱乐在线观看_日韩无码高清综合久久

鍍金池/ 問答/人工智能  Linux  網(wǎng)絡(luò)安全/ tensorflow-gpu 使用GPU時(shí),跟CPU情況下效率不變問題

tensorflow-gpu 使用GPU時(shí),跟CPU情況下效率不變問題

設(shè)置使用GPU,但是計(jì)算速度沒有什么變化,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")

圖片描述

圖片描述

回答
編輯回答
旖襯
  1. 任務(wù)管理器默認(rèn)顯示3D、Copy這幾項(xiàng),但是在用cuda的時(shí)候,主要使用的是Compute和Copy,你要切換一下
  2. Copy只有2%左右,數(shù)據(jù)量不大,模型太簡單也發(fā)揮不了GPU優(yōu)勢(shì)
2017年2月28日 13:44
編輯回答
解夏

那看你在什么數(shù)據(jù)上跑了,要是小數(shù)據(jù),本來就不需要多少計(jì)算量,反而數(shù)據(jù)在CPU和GPU之間來回傳輸還會(huì)拖慢計(jì)算。小數(shù)據(jù)量完全沒必要用GPU,發(fā)揮不出計(jì)算優(yōu)勢(shì)。

2018年7月6日 19:37