神器！AI硬件基准测试库发布

标签：#人工智能# 时间：2019/06/30 21:14:31 作者：小木

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

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

同时，作者也给出了一些测试结果。非常有意思：
<center>![](http://www.datalearner.com/resources/blog_images/f2d259c3-69c9-47f6-9cbc-cb0d6cd2c1a3.png)</center>
<center></center>

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

使用方法如下：
```python
from ai_benchmark import AIBenchmark
results = AIBenchmark().run()
```

我自己也测试了一下，非常容易：

<center>![](http://www.datalearner.com/resources/blog_images/c251f783-c5a8-4b76-b68b-70b29241bb08.png)</center>

神器！AI硬件基准测试库发布

欢迎大家关注DataLearner官方微信，接受最新的AI技术推送

相关博客

最热博客