3090双精度深度学习势能性能测试
本文于2021年6月3日发表于知乎,查看原文
本文于2021年6月3日发表于微信公众号,查看原文
组里拨科研经费新入了16块3090，正好用3090测试一下深度学习势能+分子动力学模拟（MD）的性能。
程序版本：DeePMD-kit 2.0.0.b0 双精度[1][2]，TensorFlow 2.4.1，CUDA Toolkit 11.1，LAMMPS 29Oct2020
测试结果
测试样本为examples中的水，模型为se_e2_a。
| 项目 | 3090 | 2080Ti | V100 | P100 |
|---|---|---|---|---|
| 训练10000步时间(s) | 341 | 670 | 214 | 228 |
| 10000步MD时间(s) | 129 | 154 | 49 | 54 |
可以看出，在上述测试案例中，3090的双精度性能较2080Ti有较大提升，但仍落后于V100和P100等专业计算显卡。当然，考虑到计算卡的价格也是3090的数倍，3090仍是不错的选择。