Ubuntu theano cuda 配置 之 theanorc文件 Mark

版权声明:本文为博主原创文章,未经博主允许不得转载。 https://blog.csdn.net/sinat_31824577/article/details/53140406

一.基本配置

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

阅读更多
想对作者说点什么? 我来说一句

没有更多推荐了,返回首页