AI Benchmark Alpha 是一个开源 Python 库,用于评估各种硬件平台(包括 CPU、GPU 和 TPU)的 AI 性能。该基准测试依赖于 TensorFlow 机器学习库,为评估关键深度学习模型的推理和训练速度提供轻量级且准确的解决方案。
AI Benchmark 总共包含42 项测试和19 个部分,如下所示:
MobileNet-V2 [classification]
Inception-V3 [classification]
Inception-V4 [classification]
Inception-ResNet-V2 [classification]
ResNet-V2-50 [classification]
ResNet-V2-152 [classification]
VGG-16 [classification]
SRCNN 9-5-5 [image-to-image mapping]
VGG-19 [image-to-image mapping]
ResNet-SRGAN [image-to-image mapping]
ResNet-DPED [image-to-image mapping]
U-Net [image-to-image mapping]
Nvidia-SPADE [image-to-image mapping]
ICNet [image segmentation]
PSPNet [image segmentation]
DeepLab [image segmentation]
Pixel-RNN [inpainting]
LSTM [sentence sentiment analysis]
GNMT [text translation]
更多信息和结果请访问项目网站:http://ai-benchmark.com/alpha
sudo docker build -t odyssee/ai-benchmark:latest"
sudo docker run --gpus all -it odyssee/ai-benchmark:latest"