| OS | GPU Type | Python Version | PyTorch Version | ORT-training version | DeepSpeed version | torch-ort Version | Nebula Version | MCR Image Name |
|---|---|---|---|---|---|---|---|---|
| ubuntu2004 | cu117 | 3.8 | 1.13.1 | 1.18.0 | 0.14.2 | 1.17.0 | 0.16.13 | mcr.microsoft.com/aifx/acpt/stable-ubuntu2004-cu117-py38-torch1131 |
| ubuntu2004 | cu117 | 3.9 | 1.13.1 | 1.18.0 | 0.14.2 | 1.17.0 | 0.16.13 | mcr.microsoft.com/aifx/acpt/stable-ubuntu2004-cu117-py39-torch1131 |
| ubuntu2004 | cu117 | 3.10 | 1.13.1 | 1.18.0 | 0.14.2 | 1.17.0 | 0.16.13 | mcr.microsoft.com/aifx/acpt/stable-ubuntu2004-cu117-py310-torch1131 |
| ubuntu2004 | cu118 | 3.8 | 2.0.1 | 1.18.0 | 0.14.2 | 1.17.0 | 0.16.13 | mcr.microsoft.com/aifx/acpt/stable-ubuntu2004-cu118-py38-torch201 |
| ubuntu2004 | cu118 | 3.10 | 2.0.1 | 1.18.0 | 0.14.2 | 1.17.0 | 0.16.13 | mcr.microsoft.com/aifx/acpt/stable-ubuntu2004-cu118-py310-torch201 |
| ubuntu2004 | cu118 | 3.8 | 2.2.2 | 1.18.0 | 0.14.2 | 1.17.0 | 0.16.13 | mcr.microsoft.com/aifx/acpt/stable-ubuntu2004-cu118-py38-torch222 |
| ubuntu2004 | cu118 | 3.10 | 2.2.2 | 1.18.0 | 0.14.2 | 1.17.0 | 0.16.13 | mcr.microsoft.com/aifx/acpt/stable-ubuntu2004-cu118-py310-torch222 |
| ubuntu2004 | cu121 | 3.8 | 2.2.2 | 1.18.0 | 0.14.2 | 1.17.0 | 0.16.13 | mcr.microsoft.com/aifx/acpt/stable-ubuntu2004-cu121-py38-torch222 |
| ubuntu2004 | cu121 | 3.10 | 2.2.2 | 1.18.0 | 0.14.2 | 1.17.0 | 0.16.13 | mcr.microsoft.com/aifx/acpt/stable-ubuntu2004-cu121-py310-torch222 |
| ubuntu2004 | cu118 | 3.10 | 2.3.0 | 1.18.0 | 0.14.2 | 1.17.0 | 0.16.13 | mcr.microsoft.com/aifx/acpt/stable-ubuntu2004-cu118-py310-torch230 |
| ubuntu2004 | cu121 | 3.10 | 2.3.0 | 1.18.0 | 0.14.2 | 1.17.0 | 0.16.13 | mcr.microsoft.com/aifx/acpt/stable-ubuntu2004-cu121-py310-torch230 |
| ubuntu2004 | cu121 | 3.8 | 2.1.2 | 1.18.0 | 0.14.2 | 1.17.0 | 0.16.13 | mcr.microsoft.com/aifx/acpt/stable-ubuntu2004-cu121-py38-torch212 |