tflearn GPU利用率保持低位,内存高

时间:2017-09-22 14:19:16

标签: python tensorflow gpu

当我尝试在我的项目中使用with tf.device('/gpu:0'):时,GPU并不合适。正在使用。 内存很高(91%)但GPU利用率(1%)很低。 我的结论是gpu实际上并没有被使用。 我做错了什么?

代码:

def main():
    with tf.device('/gpu:0'):
        model = cnn_model.googlenet(WIDTH, HEIGHT, 3, LR,  output=8,model_name=MODEL_NAME)
        model.fit({'input': batch_x}, {'targets': batch_y}, n_epoch=1,snapshot_step=2500, show_metric=True, run_id=MODEL_NAME)

cnn_model可以找到here

未运行时:
while not running
跑步时:
while running

0 个答案:

没有答案