训练 TensorFlow 模型时 GPU 利用率低

时间:2020-12-25 10:17:47

标签: python tensorflow gpu

我正在关注此 article 和论文(在文章中链接),用于生成草图到彩色图像的 GAN。

我安装了 TensorFlow 2.4.0 以及 CUDA 11.0 和 CuDNN 8.0.4

遵循此answer

if tf.test.gpu_device_name():
    print('Default GPU Device: {}'.format(tf.test.gpu_device_name()))
else:
    print("Please install GPU version of TF")

这输出:Default GPU Device: /device:GPU:0

按照同一问题的其他答案,tf 正在检测我的 GPU。但是我的任务管理器显示 GPU 利用率非常低:

Task Manager Screenshot

我是否缺少使用 GPU 的步骤?

0 个答案:

没有答案