python 在GPU上测试Theano

Posted

tags:

篇首语:本文由小常识网(cha138.com)小编为大家整理,主要介绍了python 在GPU上测试Theano相关的知识,希望对你有一定的参考价值。

"""
RUN ON GPU
time THEANO_FLAGS=device=gpu python theano_gpu_test.py
RUNG ON CPU
time THEANO_FLAGS=device=cpu python theano_gpu_test.py
"""

from theano import function, config, shared, tensor
import numpy
import time

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

rng = numpy.random.RandomState(22)
x = shared(numpy.asarray(rng.rand(vlen), config.floatX))
f = function([], tensor.exp(x))
print(f.maker.fgraph.toposort())
t0 = time.time()
for i in range(iters):
    r = f()
t1 = time.time()
print("Looping %d times took %f seconds" % (iters, t1 - t0))
print("Result is %s" % (r,))
if numpy.any([isinstance(x.op, tensor.Elemwise) and
              ('Gpu' not in type(x.op).__name__)
              for x in f.maker.fgraph.toposort()]):
    print('Used the cpu')
else:
    print('Used the gpu')

以上是关于python 在GPU上测试Theano的主要内容,如果未能解决你的问题,请参考以下文章

Theano 的 GPU 上是不是可以进行 int 操作?

nvcc 致命:没有为选项“gpu-architecture”定义值“sm_61”,theano 错误

TensorFlow 1.0 在 Windows 上看不到 GPU(但 Theano 有)

配置安装theano环境(非GPU版)

如何让 Theano 使用 MacBook Air 中的集成 GPU?

paper 167:GPU的使用Theano之tutorial