在 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”的实例（在第 71 行中检查），它已检测到 cpu 使用情况

为什么不使用gpu？

提前致谢

Answer 1

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

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

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