当我尝试在我的项目中使用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