目前人工智能技术发展速度很快,也很吸引眼球。但是对于各种多如牛毛的方法,目前并米有一个可靠的精准的基准来衡量各项硬件在不同算法训练和推理的性能。

现在,不用愁了。国外的一个哥们, Andrey Ignatov发布了一个python库。大家可以利用这个python库测试自己硬件的性能!
AI Benchmark Alpha是一个开源python库,用于评估各种硬件平台的AI性能,包括CPU,GPU和TPU。 该基准测试依赖于TensorFlow机器学习库,并提供精确轻量级的解决方案,用于评估关键深度学习模型的推理和训练速度。 AI Benchmark目前作为Python pip包发布,可以下载到运行Windows,Linux或macOS的任何系统。
这个包在6月26日发布了两个版本,一个是0.1.0一个是0.1.1。
目前,支持如下算法的性能测试:
● Section 1: MobileNet-V2, Classification
● Section 2: Inception-V3, Classification
● Section 3: Inception-V4, Classification
● Section 4: Inception-ResNet-V2, Classification
● Section 5: ResNet-V2-50, Classification
● Section 6: ResNet-V2-152, Classification
● Section 7: VGG-16, Classification
● Section 8: SRCNN 9-5-5, Image-to-Image Mapping
● Section 9: VGG-19, Image-to

AI Benchmark Alpha是一个开源Python库,用于评估CPU、GPU和TPU的AI性能,依赖于TensorFlow。它提供了针对多个深度学习模型的推理和训练速度测试,包括分类、图像映射和语义分割等任务。目前,GeForce GTX 1080 Ti是最强桌面GPU。用户可以通过pip安装并轻松测试硬件性能。
最低0.47元/天 解锁文章
401

被折叠的 条评论
为什么被折叠?



