机器学习中AMD平台与NVIDIA平台对比

很多人想知道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

结果如下:

8492495-90458c5e48291803.png
测试结果(平均每秒学习的样本数)
展开阅读全文

没有更多推荐了,返回首页