Theano里配置GPU的新方法

最新推荐文章于 2024-09-15 19:41:36 发布

fluffyCat

最新推荐文章于 2024-09-15 19:41:36 发布

阅读量1.6k

点赞数

分类专栏：环境搭建

本文链接：https://blog.csdn.net/fluffyCat/article/details/48826071

版权

本文介绍了在遇到Theano无法使用GPU的问题时，如何通过设置环境变量来解决。内容包括检查CUDA安装，调整.theanorc.txt配置文件，以及添加cl.exe路径的步骤。

摘要由CSDN通过智能技术生成

这几天在研究pylearn2，里面有要用到GPU，之前一直很疑惑：明明显卡是支持CUDA的，CUDA也成功安装了，也可以检测得到GPU，为什么在python程序里面就是没法运用GPU

运行下面这个检测程序

# -*- coding: utf-8 -*-
"""
Created on Tue Sep 29 08:09:11 2015

@author: Administrator
"""
import theano
import theano
from theano import function, config, shared
import theano.tensor as T
import numpy
import time

vlen = 10 * 30 * 768  # 10 x #cores x # threads per core
iters = 100

rng = numpy.random.RandomState(22)
x = shared(numpy.asarray(rng.rand(vlen), config.floatX))
f = function([], T.exp(x))
print(f.maker.fgraph.toposort())
t0 = time.time()
for i in xrange(iters):
    r = f()
t1 = time.time()
#print("Looping %d times took %f seconds" % (iters, t1 - t0))
print theano.config.device
print("Result is %s" % (r,))
if numpy.any([isinstance(x.op, T.Elemwise) for x in f.maker.fgraph.toposort()