使用GPU和THAN

Question

我正在尝试使用我的gpu执行下一个代码https://github.com/erogol/KLP_KMEANS/blob/master/klp_kmeans.py

我执行：

no_FLAGS=mode=FAST\u运行，device=gpu，floatX=float32 python klp_kmeans.py在

但上面写着：

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

在做了一点调试之后，我注意到它检测到了由于类'Gemm'的实例而导致的cpu使用率（在#71行中检查）

为什么不使用gpu？在

提前谢谢

Answer 1

您确实需要联系此脚本的作者以获得支持。“Used the cpu”消息来自此脚本，而不是来自no。是作者的代码在执行检测，而检测逻辑可能有问题。在

对于Theano来说，考虑到您在启动时看到的THEANO_FLAGS和Using gpu device 0: GeForce GTX 750 Ti消息，它将对可以在GPU上运行的所有计算图使用GPU。在

你按原样运行脚本吗？如果是这样的话，看起来它的一部分是要在CPU上运行的，而其他部分是在GPU上运行的（它似乎在对两者进行速度比较）。只有那些对klp_kmeans的调用，其中use_gpu=True将在GPU上运行，因为变量的类型（例如theano.tensor.dmatrix与theano.tensor.matrix）。在