CPU与CUDA(GPU)的计算能力对比之二: Keras Resnet 运算效率比较
结论: CUDA(GPU : NVIDIA RTX2070 MQ 笔记本版本) 启动后,效率将近是 CPU 单独运行的 17倍左右 : 每个 EPOCH 运行时间平均分别为 370秒, 22秒。
运算环境:
GPU: NVIDIA RTX2070 MQ
CPU : Intel 9750 i7 六核 2.3Ghz
CUDA : 10.1 版本
cuDNN : 7.6.5
Keras : 2.4.3
OS : Windows 10
Python :3.8.5
代码链接:
https://github.com/keras-team/keras-docs-zh/blob/master/sources/examples/cifar10_resnet.md
使用 CPU (未启动CUDA功能)时,每个 EPOCH 需要将近 370 秒。
1563/1563 [==============================] - 372s 238ms/step - loss: 1.5579 - accuracy: 0.4921 - val_loss: 1.7732 - val_accuracy: 0.4694
Learning rate