satyaog / milabench

Repository of training benchmarks

Geek Repo:Geek Repo

Github PK Tool:Github PK Tool

Milabench

Documentation

The benchmark suite has been validated on the following configurations:

Python version GPU Configuration file
3.9.12 (conda) 4x NVIDIA A100 80GB config/standard.yaml
3.9.12 (conda) 4x NVIDIA RTX8000 48GB config/standard.yaml
3.9.16 (conda) 2x NVIDIA K80 config/ci.yaml
3.9.16 (conda) 2x AMD MI100 config/ci.yaml

We are working on validating it on more configurations and will update the above table as we do.

About

Repository of training benchmarks

License:MIT License


Languages

Language:Python 95.3%Language:Cuda 1.9%Language:Shell 1.7%Language:Roff 0.7%Language:C++ 0.5%