本文介绍了在滴滴云上进行的NVIDIA A100 40G GPU性能测试，对比了V100和2080Ti的深度学习能力。通过TensorFlow Benchmarks，展示了A100相对于V100的1.5到1.7倍性能提升，并探讨了MIG技术对性能的影响。测试结果显示，A100在某些任务上能实现接近两倍于V100的性能。
眼看游戏卡RTX3080 发售在即，我终于等到了滴滴云（感谢）A100的测试机会。因为新卡比较紧张，一直在排队中，直到昨天才拿了半张A100...今天终于上手了单张40G的A100，小激动，小激动，小激动！！！基于安培架构的最新一代卡皇（NVIDIA GPU A100 Ampere）可以搞起来了。
Part 1：系统环境
A100正处于内存阶段，官网上还看不到。内测通过ssh连接，ssh连上去之后大概看了下系统环境。
操作系统，CPU，RAM数据如上。重点关注GPU：A100-SXM4-40GB （上次摸DGX A100的时候，没有把测试跑起来，好悔）
CUDA11，CudNN，TensorFlow1.5.2 等配套环境滴滴云都已经部署好了，可以省去好多时间！
这里需要注意，新版显卡必须要用CUDA11，而且得用NV自己编译的TensorFlow1.5.2。
然后，网上捞一段Python代码：
from 输出：
Created TensorFlow device (/device:GPU:0 with 36672 MB memor
最低0.47元/天 解锁文章
扫一扫
被折叠的 条评论
为什么被折叠?
到【灌水乐园】发言
