在 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 上運行。