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
本文介绍Theano配置模块的使用方法,包括三种改变配置的方式:直接修改theano.config属性,通过THEANO_FLAGS环境变量,以及编辑theanorc配置文件。文章详细解释了如何通过这些方式设置floatX精度和设备类型等关键参数。
426

被折叠的 条评论
为什么被折叠?



