相关疑难解决方法(0)

在训练LSTM/RNN模型时,为什么我的GPU比CPU慢?

我的机器有以下规格:

CPU:Xeon E5-1620 v4

GPU:Titan X(Pascal)

Ubuntu 16.04

Nvidia车手375.26

CUDA取得8.0

cuDNN 5.1

我已经使用Tensorflow作为支持参考的以下Keras示例进行了基准测试:

SCRIPT NAME                  GPU       CPU
stated_lstm.py               5sec      5sec 
babi_rnn.py                  10sec     12sec
imdb_bidirectional_lstm.py   240sec    116sec
imbd_lstm.py                 113sec    106sec
Run Code Online (Sandbox Code Playgroud)

我的gpu显然在非lstm模型中执行我的cpu.

SCRIPT NAME                  GPU       CPU
cifar10_cnn.py               12sec     123sec
imdb_cnn.py                  5sec      119sec
mnist_cnn.py                 3sec      47sec 
Run Code Online (Sandbox Code Playgroud)

还有其他人经历过这个吗?

nvidia machine-learning keras tensorflow

22
推荐指数
4
解决办法
2万
查看次数

标签 统计

keras ×1

machine-learning ×1

nvidia ×1

tensorflow ×1