我的机器有以下规格:
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)
还有其他人经历过这个吗?