使用 gpu 导入 theano 失败，因为“内存不足”错误

Question

当我尝试导入 theano 库时，我会收到以下错误：

>>> import theano
ERROR (theano.sandbox.cuda): ERROR: Not using GPU. Initialisation of device gpu failed:
initCnmem: cnmemInit call failed! Reason=CNMEM_STATUS_OUT_OF_MEMORY. numdev=1

我的 .theanorc 的内容：

[global]
floatX = float32
device = gpu
optimizer = fast_run

[lib]
cnmem = 0.9

[nvcc]
fastmath = True

[blas]
ldflags = -llapack -lblas

[cmodule]
mac_framework_link=True

我也尝试过使用 theano 标志运行 python 环境，但仍然是同样的问题：

$ THEANO_FLAGS=mode=FAST_RUN,device=gpu,floatX=float32,lib.cnmem=0.9 python3

他们还在 Theano Github 页面上提到了类似的问题：Initialisation of device gpu failed!，但是他们在谈论 float 与 int 的值 CNMeM，我想我已经正确设置为 0.9（无特殊原因，我实际上尝试了1.0、0.95、0.5 和其他一些值，结果相同）。

我的设置：

• Mac OS X 10.11.4
• Xcode 7.2（我不得不从 7.3 降级）
• 库达 7.5.19
• Python 3.5.1

解决方案：

首先，我没有意识到CNMeM 的值是占总内存的百分比，而且因为我使用的是带有两个显示器的 gpu，我实际上并没有太多可用的内存，无论如何，我已经尝试过 0.1 并且它有效。

我通过将这一行添加到 theano 配置解决了禁用 cuDNN 的第二个问题：

optimizer_including = cudnn

查看此页面了解更多详情：sandbox/cuda/dnn

Answer 1

首先，我没有意识到CNMeM 的值是占总内存的百分比，而且因为我使用的是带有两个显示器的 gpu，我实际上并没有太多可用的内存，无论如何，我已经尝试过 0.1 并且它有效。

我通过将这一行添加到 theano 配置解决了禁用 cuDNN 的第二个问题：

optimizer_including = cudnn

查看此页面了解更多详情：sandbox/cuda/dnn