内容简介:目前人工智能技术发展速度很快,也很吸引眼球。但是对于各种多如牛毛的方法,目前并米有一个可靠的精准的基准来衡量各项硬件在不同算法训练和推理的性能。现在,不用愁了。国外的一个哥们, Andrey Ignatov发布了一个python库。大家可以利用这个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
同时,作者也给出了一些测试结果。非常有意思:
目前最强的桌面GPU当属于GeForce GTX 1080 Ti了。其次是TITAN Xp CE和GeForce GTX TITAN X。
使用这个库也很简单,大家可以先pip install ai-benchmark。注意,需要安装tensorflow才能运行。
使用方法如下:
from ai_benchmark import AIBenchmark results = AIBenchmark().run()
我自己也测试了一下,非常容易:
可以看到,我的硬件在MobieNet-V2算法的训练速度大约是27688±741ms,推理速度大约是2747±119ms。这速度惨不忍睹啊。各位可以自己去看看自己的成绩。
以上就是本文的全部内容,希望本文的内容对大家的学习或者工作能带来一定的帮助,也希望大家多多支持 码农网
猜你喜欢:本站部分资源来源于网络,本站转载出于传递更多信息之目的,版权归原作者或者来源机构所有,如转载稿涉及版权问题,请联系我们。
宇宙涟漪中的孩子
谢云宁 / 四川科学技术出版社 / 2017-11 / 28.00元
近未来。日冕科技公司通过建造围绕太阳的光幕搜集了近乎无穷的能源,这些能源主要用于地球上的网络空间建设。随着全球网络时间频率的不断提升,越来越多的人选择接驳进虚拟空间,体验现实中难以经历的丰富人生。 网络互动小说作者宁天穹一直自认为是这些人中普通的一员,有一天却被一名读者带进反抗组织,了解到日冕公司的各种秘密,并被告知自己的小说将在抵抗运动中起到重要作用。 起初他拒绝参与,但看到地球被笼......一起来看看 《宇宙涟漪中的孩子》 这本书的介绍吧!