theano的config模块包含了许多属性来定义theano的动作。改变theano配置有三种方式,按照优先级依次为:
- theano.config.property
- 赋值THEANO_FLAGS,通过THEANO_FLAGS来改变配置。
THEANO_FLAGS=’floatX=float32,device=cuda0,lib.cnmem=1’ python .py - 通过修改./theanorc来改变配置,在linux下面是$HOME/.theanorc。
生成文件 sudo gedit ~/.theanorc(注意不要漏掉theano前面的一个点)并复制如下内容,然后保存,其中cuda一项的内容为cuda所安装的位置。
[global]
floatX=float32
device=cuda
[nvcc]
flags=-D_FORCE_INLINES
参考:http://deeplearning.net/software/theano/library/config.html