python在gpu上运行_python – 如何在GPU上运行theano

如果我使用

python 3.5运行以下代码

import numpy as np

import time

import theano

A = np.random.rand(1000,10000).astype(theano.config.floatX)

B = np.random.rand(10000,1000).astype(theano.config.floatX)

np_start = time.time()

AB = A.dot(B)

np_end = time.time()

X,Y = theano.tensor.matrices('XY')

mf = theano.function([X,Y],X.dot(Y))

t_start = time.time()

tAB = mf(A,B)

t_end = time.time()

print ("NP time: %f[s], theano time: %f[s] **(times should be close when run

on CPU!)**" %(np_end-np_start, t_end-t_start))

print ("Result difference: %f" % (np.abs(AB-tAB).max(), ))

我得到了输出

NP time: 0.161123[s], theano time: 0.167119[s] (times should be close when

run on CPU!)

Result difference: 0.000000

它说,如果时间接近,这意味着我在我的CPU上运行.

如何在GPU上运行此代码？

注意：

>我有一台配有Nvidia Quadro k4200的工作站.

>我已经安装了Cuda工具包

>我在VS2012上成功完成了cuda vectorAdd示例项目.