「大神器!」硬件的AI性能测试Python库发布

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

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

[[269231]]

现在,不用愁了。国外的一个哥们, 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-Image Mapping

● Section 10: ResNet-SRGAN, Image-to-Image Mapping

● Section 11: ResNet-DPED, Image-to-Image Mapping

● Section 12: U-Net, Image-to-Image Mapping

● Section 13: Nvidia-SPADE, Image-to-Image Mapping

● Section 14: ICNet, Image Segmentation

● Section 15: PSPNet, Image Segmentation

● Section 16: DeepLab, Image Segmentation

● Section 17: Pixel-RNN, Image Inpainting

● Section 18: LSTM, Sentence Sentiment Analysis

● Section 19: GNMT, Text Translation

同时,作者也给出了一些测试结果。非常有意思:

「大神器!」硬件的AI性能测试Python库发布

目前***的桌面GPU当属于GeForce GTX 1080 Ti了。其次是TITAN Xp CE和GeForce GTX TITAN X。

使用这个库也很简单,大家可以先pip install ai-benchmark。注意,需要安装tensorflow才能运行。

使用方法如下:

  1. from ai_benchmark import AIBenchmark 
  2. results = AIBenchmark().run() 

我自己也测试了一下,非常容易:

「大神器!」硬件的AI性能测试Python库发布

可以看到,我的硬件在MobieNet-V2算法的训练速度大约是27688±741ms,推理速度大约是2747±119ms。这速度惨不忍睹啊。各位可以自己去看看自己的成绩。

责任编辑:张燕妮 来源: 数据学习DataLearner
相关推荐

2020-09-29 07:38:22

Python装饰器框架

2011-09-19 13:11:00

Vista性能测试

2024-05-22 12:07:12

向量数据库AI

2019-06-18 10:24:23

开源技术 趋势

2023-12-06 07:36:27

前端开发

2010-09-16 11:09:34

IE 9 Beta评测

2021-01-31 20:51:55

PuppeteerNode核心

2021-05-27 12:10:42

前端puppeteer代码

2021-09-13 09:28:10

PuppeteerNode 库DevTools 协议

2024-09-06 07:55:42

2009-07-31 16:29:47

ibmdwXML

2018-06-08 10:18:22

Python装饰器迭代器

2012-06-28 10:18:01

数据库

2022-11-27 08:12:11

RocketMQ源码工具类

2023-03-07 10:45:31

AI自动化测试

2017-01-05 14:01:38

linux密码高强度

2016-09-13 19:21:07

CTO管理技术

2010-02-02 13:20:24

以太网交换机

2011-03-10 14:11:21

GPU硬件加速IE9

2024-05-15 09:05:43

Python文档处理工具自动化文档生成
点赞
收藏

51CTO技术栈公众号