python - 在 Theano 中使用 GPU

Question

我正在尝试使用我的 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

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

为什么不使用gpu？

提前致谢

Answer 1

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

就 Theano 而言，鉴于您THEANO_FLAGS和Using gpu device 0: GeForce GTX 750 Ti您在启动时看到的“”消息，它将使用 GPU 来处理所有可以在 GPU 上运行的计算图。

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