儘管安裝了tensorflow-GPU，GPU仍未用於計算

我的電腦安裝了以下軟件：Anaconda（3），TensorFlow（GPU）和Keras。有兩個Anaconda虛擬環境 - 一個用於Python 2.7的TensorFlow和一個用於3.5的GPU，版本爲installed according to the TF instructions。（我曾在一個單獨的環境中以前安裝TensorFlow的CPU版本，但我已經刪除了它。）儘管安裝了tensorflow-GPU，GPU仍未用於計算

當我運行以下命令：

source activate tensorflow-gpu-3.5 
python code.py

，並檢查nvidia-smi它只顯示3MiB GPU內存使用通過Python，所以它看起來像GPU不用於計算。（code.py是一個簡單的深度Q學習算法與凱拉斯實施）

任何想法可能會出錯？

來源

2017-05-06 Massyanya

我的GPU未運行，是因爲一個破碎的安裝CuDNN的原因，更精確地庫和來源不同版本CuDNN來了。

它被修復了the following piece of advice.

來源

2017-05-28 15:57:11 Massyanya

調試這些問題的好方法是檢查哪些操作已分配給哪些設備。

你可以通過配置參數的會話檢查：

session = tf.Session(config=tf.ConfigProto(log_device_placement=True))

當你運行你的應用程序，你會看到一些輸出指示正在使用的設備。

你可以在這裏找到更多的信息： https://www.tensorflow.org/tutorials/using_gpu

來源

2017-05-06 17:18:46 purpletentacle

儘管安裝了tensorflow-GPU，GPU仍未用於計算

回答

相關問題