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

作者: 小木 2019-07-01 08:46:37

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

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

现在,不用愁了。国外的一个哥们, 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。这速度惨不忍睹啊。各位可以自己去看看自己的成绩。

AI 数据 人工智能
上一篇:中国发布信息技术十大前沿热点问题,自动驾驶等入选 下一篇:百度王海峰携开源开放的深度学习平台飞桨亮相2019软博会
评论
取消
暂无评论,快去成为第一个评论的人吧

更多资讯推荐

德媒:欧盟拟立法限制滥用人工智能

据德国《法兰克福汇报》网站4月13日报道,人工智能的胜利前进已不可阻挡。新冠疫情尤其让人们关注到这种拥有自我学习能力的系统对医疗体系组织工作的价值。

参考消息 ·  11h前
国内首个!北京拟推进自动驾驶商业化,年内将实现真无人驾驶

乘自动驾驶出租车要付费,无人配送车街上跑,路测拿掉安全员,无人驾驶车上高速……这些即将在北京实现。

南方都市报 ·  13h前
前沿洞察丨无人机送货不迷路的原因竟在这里!

本期前沿洞察为大家带来这些技术:用微观交叉定位,让无人机送货不再找路难;能暴露行动轨迹的智能袜子;基于两束交叉光触发的化学反应实现微米级高精度3D打印......一起来看看吧!

望潮科技 ·  13h前
2021年AI智能摄像机带来的新市场

大流行除了给全球经济带来巨大影响之外,也加速了越来越多的先进技术走向成熟应用,如人工智能(AI)和机器学习(ML),技术时代的到来往往伴随着人类的迫切需求。

蒙光伟 ·  23h前
OpenAI CEO Sam Altman:AI革命即将到来,我们需要新的系统

我们正处于这场技术变革的开端,我们拥有创造未来的宝贵机会。而这不是简单地解决目前的社会和政治问题,它必须为完全不同的社会而设计。

Sam Altman ·  1天前
谈谈基于深度学习的目标检测网络为什么会误检,以及如何优化目标检测的误检问题

在训练人脸检测网络时,一般都会做数据增强,为图像模拟不同姿态、不同光照等复杂情况,这就有可能产生过亮的人脸图像,“过亮”的人脸看起来就像发光的灯泡一样。

刘冲 ·  1天前
值得思考:197亿美金,微软2021年的AI转型之路

4月12日,微软宣布将以每股56美元的价格收购语音识别巨头Nuance,出价达到了197亿美元。

东方林语 ·  1天前
人工智能优先战略将从哪里开始?

人工智能可以为企业带来竞争优势,并释放难以获得的巨大商机。因此,人们需要了解制定有效的人工智能优先策略的6个步骤。

李睿 ·  1天前
Copyright©2005-2021 51CTO.COM 版权所有 未经许可 请勿转载