如何设置theano配置

时间:2014-02-06 16:07:52

标签: theano

我是Theano的新手。 试图设置配置文件。

首先,我注意到我没有.theanorc文件:

  1. locate .theanorc - 不返回任何内容
  2. echo $THEANORC - 不返回任何内容
  3. theano.test() - 传递确定
  4. 我猜我安装了theano时创建了一些默认配置。它在哪里?

5 个答案:

答案 0 :(得分:43)

Theano本身不会创建任何配置文件,但其所有配置标志都有默认值。如果要修改默认值,则只需要这样的文件。

这可以通过在主目录中创建.theanorc文件来完成。例如,如果您希望floatX始终为float32,则可以执行以下操作:

echo -e "\n[global]\nfloatX=float32\n" >> ~/.theanorc

在Linux和Mac下。在Windows下,这也可以完成。有关详细信息,请参阅此页:

http://deeplearning.net/software/theano/library/config.html

答案 1 :(得分:7)

在Linux终端主目录中写:

nano .theanorc

在文件中复制以下行

[global]
floatX = float32
device = gpu0

[lib]
cnmem = 1   

保存。

当我在python中导入theano时,我遇到了cnmem内存问题。似乎是因为监视器连接到gpu。要解决它,请将cnmem更改为0.8。这个数字低于1是为theano保留的gpu的百分比

答案 2 :(得分:3)

我有一个类似的问题,这对我有帮助:

import theano
//...
theano.config.floatX = 'float32' //or 'float64' whatever you want

答案 3 :(得分:1)

这对我有用:

nano ~/.theanorc

然后我进入了:

[global]
floatX = float32
device = cuda

检查Theano是否使用GPU的代码位于Theano doc page

(我使用的是Ubuntu 14.04,Theano 0.9.0(conda),NVIDIA 1080 Ti GPU)。

答案 4 :(得分:0)

我遇到过类似的问题。我在配备华硕Z270E主板的台式机上安装了NVIDIA 1070 GPU,并在设置.theanorc文件后能够导入theano。 (然后重新启动)

[global]
floatX = float32
device = gpu

[cuda]
root = /usr/local/cuda
[lib]
cnmem = 1