在GPU(Windows)上运行python / theano脚本

时间:2019-08-05 18:51:11

标签: python gpu theano theano-cuda

我想要的

要运行如下所示的简单python脚本,但我希望它使用CPU而不是CPU:

import theano

def some_convolutional_neural_network():
    #Insert (theano) code here

我想要的是关于如何实现此目标的全面说明,因为theano文档非常复杂。对于我和以后的读者来说,如果有这样的启发式描述,那将是很好的。

我尝试过的事情

首先,我在Windows上遵循Installation instructions。然后,我尝试使theano使用我的GPU。

我找到了部分答案here,看来解决方法是修改环境变量THEANO_FLAGS或名为.theanorc的配置文件是配置文件。但是,documentation对我来说还是不清楚的,我无法在任何地方找到环境var或配置文件。任何帮助将不胜感激,因为我宁愿停留在这里。

我的规格:  我在Windows 10上,并且有GeForce GTX 650 Ti。

0 个答案:

没有答案