基于keras的GPU机器学习的负载测试

时间:2018-09-23 13:45:11

标签: python keras

我正在寻找一个简单的(自包含)基于keras的示例,该示例使我的GPU长时间运行并具有最大的使用率。

以此,我想测试机器的热性能。

1 个答案:

答案 0 :(得分:0)

我不确定,但是您可以尝试遵循此Keras tutorial,而不是对每个类使用1K的图像,而是使用整个数据集,对每个类使用12,5K的图像。为了确保您正在使用整个可用的视频内存,请增加批处理大小,直到在运行模型时开始收到警告为止。一个好的起点是批处理大小为64。尝试执行此操作,如果未收到任何警告,则将其增加一些。比方说80。然后再次运行模型。根据我的经验,我认为“正确”的批次将在64-128之间。

P.S。:我不记得确切的警告消息,但是您一定会在看到警告消息时知道这一点