Ubuntu16.04 Theano报错Theano: CNMe

现象

在python中运行Theano报错

import theano

Theano: CNMeM is disabled, CuDNN not available

原因

Cuda中没有安装管理内存的包

解决办法

下载编译cnmem包，并放到cuda安装目录下。

git clone https://github.com/NVIDIA/cnmem.git
mkdir build
cd build
cmake ..
make
sudo cp libcnmem.so /usr/local/cuda/lib64
sudo cp libcnmem.so.1 /usr/local/cuda/lib64
sudo cp libcnmem.so.1.0.0 /usr/local/cuda/lib64
sudo cp ../include/cnmem.h /usr/local/cuda/include

在~/.theanorc文件中添加如下信息：

[lib]
cnmem=0.8