一.基本配置
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