使用GPU与不使用GPU的结果有何不同?

时间:2019-09-21 03:59:58

标签: python-3.x keras gpu conv-neural-network

我有一个带有2个隐藏层的CNN。当我在具有8GB RAM的cpu上使用keras时,有时会出现“ Memory Error”或有时精度级别为0,但同时某些级别为1.00。如果我在GPU上使用keras,是否可以解决我的问题?

1 个答案:

答案 0 :(得分:0)

在训练期间,您可能没有足够的内存来容纳CPU中的所有图像。仅在拥有更多内存的情况下使用GPU才有帮助。如果由于图像过多或分辨率太高而发生这种情况,则可以尝试使用keras的ImageDataGenerator和任何流方法批量添加数据。