机器学习中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

结果如下：

测试结果（平均每秒学习的样本数）
最后编辑于 ：
©著作权归作者所有,转载或内容合作请联系作者
平台声明：文章内容（如有图片或视频亦包括在内）由作者上传并发布，文章内容仅代表作者本人观点，简书系信息发布平台，仅提供信息存储服务。

推荐阅读更多精彩内容