前提

  • Theano 1.0.0
  • Cmake 3.11.1
  • Cuda 8

第一步 配置 .theanorc

在根目录下配置theano配置文件 .theanorc

1
2
cd 
vim .theanorc

.theanorc内容如下,保存退出

1
2
3
4
5
6
7
8
9
[cuda]
root = /usr/local/cuda/

[global]
floatX = float32
device = cuda

[lib]
cnmem = 1.0

判断是否配置正确,输入python进入python交互环境,输入import theano
若提示No module nameed pygpu,则进入第二步。
若没有提示使用Cuda或者GPU的信息,也进入第二步。

第二步 安装 gpuarray

1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
cd <dir>
rm -rf ~/.local/lib/libgpuarray* ~/.local/include/gpuarray
rm -rf build Build
mkdir Build
cd Build
cmake .. -DCMAKE_INSTALL_PREFIX=~/.local -DCMAKE_BUILD_TYPE=Release
make
make install
DEVICE="<test device>" make test

cd ..

# Run the following export and add them in your ~/.bashrc file
export CPATH=$CPATH:~/.local/include
export LIBRARY_PATH=$LIBRARY_PATH:~/.local/lib
export LD_LIBRARY_PATH=$LD_LIBRARY_PATH:~/.local/lib

python setup.py build
python setup.py install --user
cd
DEVICE="<test device>" python -c "import pygpu;pygpu.test()"

参考资料