Question

我尝试使用my gpu

执行下一个代码https://github.com/erogol/KLP_KMEANS/blob/master/klp_kmeans.py

我执行：

THEANO_FLAGS = mode = FAST_RUN，device = gpu，floatX = float32 python klp_kmeans.py

但它说：

Using gpu device 0: GeForce GTX 750 Ti
Cluster number comparison ----
Used the cpu

在做了一点调试之后，我注意到它已经检测到由于类的实例而导致的cpu使用情况＆＃39; Gemm＆＃39; （在第71行检查）

为什么不使用gpu？

提前致谢

Answer 1

您确实需要与此脚本的作者联系以获取支持。＆＃39; Used the cpu＆＃39;消息来自这个脚本，而不是来自Theano。它是正在进行检测的作者代码，检测逻辑可能有问题。

就Theano而言，鉴于您的THEANO_FLAGS和＆＃39; Using gpu device 0: GeForce GTX 750 Ti＆＃39;你在启动时看到的消息，它会将GPU用于可以在GPU上运行的所有计算图。

您是否按原样运行脚本？如果是这样，看起来它的一部分打算在CPU上运行，而GPU上的其他部分（它似乎在两者之间进行速度比较）。由于输入变量的方式（例如klp_kmeans与use_gpu=True），只有那些theano.tensor.dmatrix theano.tensor.matrix的调用将在GPU上运行。

与Theano一起使用GPU

1 个答案: