【发布时间】:2018-04-05 00:04:24
【问题描述】:
我正在尝试使用 3 个类和每个类中大约 500 个图像来训练一个 CNN 模型。图像大小为 [965*995*1]。当我运行代码时,我收到一条错误消息,提示 “超出了设备上允许的最大变量大小。”
模型给出错误“设备内存不足。要查看有关 GPU上的可用内存的更多详细信息,请使用'gpuDevice()'。如果问题仍然存在,当我将图像大小调整为 [128 *128] 时, 通过调用“gpuDevice(1)”重置 GPU。
当我将图像大小调整为 [64*64] 时,它运行良好。
我该如何解决这个问题?
注意:系统有 16GB 的 RAM。
【问题讨论】:
标签: matlab