編譯器在編譯時會自動使用GPU還是必須手動使用它？

Question

我一直在用8GB RAM（英特爾酷睿i7-4th gen）在我的PC上編譯該程序以進行鳥類圖像預處理 。 處理器，但每次運行一段時間后都會卡住，顯示此錯誤：

File "bird.py", line 13, in <module>
    X,Y,X_test,Y_test = pickle.load(open("full_dataset.pkl", "rb"))
MemoryError

作為tensorflow的新手，我不確定是否正在使用CUDA（已成功安裝）或只是CPU仍在執行計算。

Answer 1

在發出任何GPU計算（或者實際上是任何TensorFlow計算）之前，都會引發您所看到的錯誤。 MemoryError表示取消挑選數據集可能導致內存不足，或者數據中存在某些損壞。 嘗試加載較小的數據集，以查看問題是否仍然存在。

關於您的原始問題，如果您已經安裝了TensorFlow的GPU版本以及兼容的CUDA版本（即CUDA 8.0）和cuDNN，TensorFlow會將許多計算任務轉移到您的GPU上，包括將要構成的矩陣乘法和卷積培訓期間花費的大部分時間。