我目前正在 tensorflow GPU 上对 3D 医学图像运行 CNN。每当我运行代码时，命令提示符上都会出现资源耗尽错误。我已经尝试过以 1 号的小批量运行代码（一次一名患者）。
我的 GPU 是 NVIDIA GeForce GTX 960。我正在查看我的 GPU 的规格，但我不确定哪个组件限制了内存。它是标准内存配置（2GB）吗？
命令提示符返回以下内容：
2017-06-11 16:23:37.095587: W c:\tf_jenkins\home\workspace\release-win\device\gpu\os\windows\tensorflow\core\framework\op_kernel.cc:1152] 资源耗尽：OOM 时使用 shape[3,3,3,8,16] 分配张量 2017-06-11 16:23:47.096178: W c:\tf_jenkins\home\workspace\release-win\device\gpu\os\windows\tensorflow\ core\common_runtime\bfc_allocator.cc:273] 分配器 (GPU_0_bfc) 试图分配 365.63MiB 时内存不足。
当前分配摘要如下。2017-06-11 16:23:47.096349: 我 c:\tf_jenkins\home\workspace\release-win\device\gpu\os\windows\tensorflow\core\common_runtime\bfc_allocator.cc:643] Bin (256)：总块：0，正在使用的块：0 0B 分配给块。0B 客户端请求的块。0B 在 bin 中使用。0B 客户端请求在 bin 中使用。...... 2017-06-11 16:23:47.144036: I c:\tf_jenkins\home\workspace\release-win\device\gpu\os\windows\tensorflow\core\common_runtime\bfc_allocator.cc:696 ] 3 个大小为 13824 的块，总计 40.5KiB 2017-06-11 16:23:47.144745: I c:\tf_jenkins\home\workspace\release-win\device\gpu\os\windows\tensorflow\core\common_runtime\bfc_allocator。 cc:696] 3 个大小为 383385600 的块，总计 1.07GiB 2017-06-11 16:23:47.145486: I c:
统计：
限额：1548396134
使用中：1548374272
最大使用量：1548396032
NumAllocs：35
最大分配大小：398141184