很多人想知道AMD的ROCm平台和NVIDA的CUDA平台在深度学习中哪个速度更快。我没有NVIDIA的GPU,但是最近看到了一篇这两个平台对比的文章,我简单介绍一下。
原文链接:Benchmark CIFAR10 on TensorFlow with ROCm on AMD GPUs vs CUDA9 and cuDNN7 on NVIDIA GPUs
该文章使用的是CIFAR10训练集在AMD和NVIDIA的多个GPU平台上进行学习,并对比测试结果。
平台如下:
AMD(TF1.0.1):
Ubuntu 16.04.3 x64
HIP-TensorFlow 1.0.1
Python 2.7
Driver: ROCm 1.7
AMD(TF1.3):
Ubuntu 16.04.4 x64
TensorFlow 1.3
Python 3.5
Driver: ROCm 1.7.137
NVIDIA(TF1.6):
Ubuntu 16.04.4 x64
TensorFlow r1.6
Python 3.5
Driver: 390.30, CUDA9.0, cuDNN7
结果如下: