时间戳

首页 > 解决方案 > keras Deep Learning Slowness - 使用 Python Chollet 进行深度学习的示例 IMDB 数据集

python - keras Deep Learning Slowness - 使用 Python Chollet 进行深度学习的示例 IMDB 数据集

问题描述

我遇到了 keras 问题，导致我的处理器在处理示例时似乎陷入了困境。

例如，在 IMDB 数据集中（如果有人知道这本书，请执行 Chollet 在 Python 深度学习中的练习 3.4.1），运行脚本：

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

产生类似于以下内容的输出：

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

随着数字变大并接近完成，这种更新变得越来越慢。

我假设我安装的 keras/Tensorflow/CUDA/cuDNN 是罪魁祸首，但很好奇您是否知道任何可以解决问题的明显内容。

运行 Ubuntu Linux、NVIDIA GTX 1080、Keras/Tensorflow (GPU)/CUDA、cuDNN（也许，假设我正确安装了所有可能不准确的东西）。

谢谢！

标签： pythontensorflowkerasdeep-learning

解决方案

推荐阅读