将GPU用于视频数据集时出现内存不足错误

时间:2020-10-27 21:04:12

标签: keras out-of-memory gpu

我的数据集大小约为5GB,包括2000个短视频。我正在使用带有8个32GB GPU的系统。 但是,当我在纪元7中运行代码时(有时也发生在6,8,9中)。 This is the error message screen shot 如图所示,该错误发生在GPU [0]中,而我们有8个GPU,(GPU [0],GPU 1,GPU [2],GPU [3],GPU [4],GPU [5 ],GPU [6],GPU [7])。我不知道服务器是顺序使用GPU还是并行使用GPU。我应该如何首先定义并激活GPU以摆脱此错误消息? 请帮助我解决内存不足的错误。

谢谢

0 个答案:

没有答案