问题描述
我正在寻找一个简单的（自包含）基于keras的示例，该示例使我的GPU长时间运行并具有最大的使用率。
借此，我想测试机器的热性能。
1楼
我不确定，但是，您可以尝试按照此 ，而不是对每个类使用1K的图像，而是使用整个数据集，每个类使用12,5K的图像。 为了确保您正在使用整个可用的视频内存，请增加批处理大小，直到在运行模型时开始收到警告为止。 一个好的起点是批处理大小为64。 尝试执行此操作，如果未收到任何警告，则将其增加一些。 比方说80。然后再次运行模型。 根据我的经验，我认为“正确”的批次将在64-128之间。
PS：我不记得确切的警告消息，但是您肯定会在看到警告消息时知道它