keras Deep Learning Slowness - 來自 Deep Learning with Python Chollet 的示例 IMDB 數據集

Question

我遇到 keras 問題，導致我的處理器在處理示例時似乎陷入困境。

例如在 IMDB 數據集中（深度學習中的練習 3.4.1，Chollet 的 Python，如果有人知道這本書的話），運行腳本：

import keras
from keras.datasets import imdb    
(train_data, train_labels), (test_data, test_labels) = 
    imdb.load_data(num_words=10000)

產生一個 output 看起來像：

[=====>...] - ETA: 59s✓✓✓✓✓✓✓✓✓✓✓✓✓✓✓✓✓✓✓✓✓✓16105472/17464789

隨着數字變大並接近完成，更新速度越來越慢。

我假設我安裝的 keras/Tensorflow/CUDA/cuDNN 是罪魁禍首，但很好奇您是否知道任何可以解決問題的明顯方法。

運行 Ubuntu Linux、NVIDIA GTX 1080、Keras/Tensorflow (GPU)/CUDA、cuDNN（也許，假設我正確安裝了所有東西，這可能不准確）。

謝謝！

Answer 1

此進度條在第一次初始下載期間顯示，不應出現在隨后的數據導入中。 可能有幾個問題導致速度變慢和/或失敗：

• 您的互聯網連接不穩定。
• 文件服務存在問題。 也許存儲庫服務器提供損壞的文件？ （您可以嘗試強制從另一個存儲庫下載，請參閱How to setup pip to download from mirror repository by default? ）
• 您的本地光盤或以前的部分下載是正確的：您可以嘗試刪除 ~/.keras/datasets/mnist.npz 中的部分下載
• 檢查您的硬盤是否已滿。