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"