Ubuntu theano cuda 配置 之 theanorc文件 Mark

一.基本配置

ctrl+alt+T打开终端，输入:

sudo gedit ~/.theanorc

在空白文件里写入：

[global] 
floatX=float32 
device=cpu

保存即可使用theano

二.其他功能

1.使用GPU

device=gpu 

[cuda]
root=/usr/local/cuda-8.0

2.使用cudnn加速

[nvcc]
fastmath = True
optimizer_including=cudnn

3.保证使用cuda运行过程中参数不随机

[dnn.conv]
algo_bwd_filter = deterministic
algo_bwd_data = deterministic

4.并行操作配置

默认情况下，显卡显存占满，只允许一个程序运行如下所示：初始化时调用30%的显存，可最大允许三个程序同时运行。

[lib]
cnmem=0.3

三.汇总

[global]  
device=gpu  
floatX=float32 

[dnn.conv]
algo_bwd_filter = deterministic
algo_bwd_data = deterministic

[cuda]
root=/usr/local/cuda-8.0

[lib]
cnmem=0.3

[nvcc]
fastmath = True
optimizer_including=cudnn

【参考】

http://deeplearning.net/software/theano/library/config.html
https://zhuanlan.zhihu.com/p/23250782