Tensorflow 在家用 PC 上的 GPU 上運行，但不在 Google colab 上運行

Question

我正在嘗試在 google colab 上的這個 repo 中訓練神經網絡： https : //github.com/carpedm20/DCGAN-tensorflow

我在 Google Colab 上運行此代碼（硬件加速器設置為 GPU），訓練速度比在 GTX 1050 弱得多的家用 PC 上運行慢 40 倍。所以我猜在 colab 上它沒有運行在 GPU 上。 google colab 上的輸出確實表明找到了 GPU：

在以下位置找到 GPU：/device:GPU:0

有誰知道這是為什么？

Answer 1

我認為這些數據背后的原因是需要時間從驅動程序到 colab 嘗試並行功能上的 GPU 100 次，這兩者都不需要數據來運行並告訴我結果

畢竟，當我嘗試時，我發現可以在 39 秒內完成第一個 Epoch 的示例，似乎 colab 需要時間為我擁有的每張照片傳輸數據，並在初始操作后的第一個 Epoch 332s 中每次您將運行他都會得到 39 秒，我將得到 332 秒

我希望這有幫助

Answer 2

就我而言，在Google Colab上加載數據存在問題。 即使我們使用GPU，也會極大地影響速度。