Skip to content

Commit

Permalink
Update image list
Browse files Browse the repository at this point in the history
  • Loading branch information
@sophies927
sophies927 committed May 30, 2024
1 parent 742aad5 commit ef5408e
Show file tree
Hide file tree
Showing 2 changed files with 15 additions and 30 deletions.
17 changes: 1 addition & 16 deletions docs/ecosystem/acpt.md
View file
Original file line number Diff line number Diff line change
Expand Up @@ -25,22 +25,7 @@ Azure Container for PyTorch (ACPT) is a lightweight, standalone environment that
* **Latest training optimization technologies:** Make use of ONNX Runtime, DeepSpeed, MSCCL, and more.

## Supported configurations for Azure Container for PyTorch (ACPT)
The following configurations are supported in the Microsoft Container Registry (MCR):

| OS | GPU Type | Python Version | PyTorch Version | ORT-training version | DeepSpeed version | torch-ort Version | Nebula Version |
| - | - | - | - | - | - | - | - |
|ubuntu2004|cu117|3.8|1.13.1|1.18.0|0.14.2|1.17.0|0.16.13|
|ubuntu2004|cu117|3.9|1.13.1|1.18.0|0.14.2|1.17.0|0.16.13|
|ubuntu2004|cu117|3.10|1.13.1|1.18.0|0.14.2|1.17.0|0.16.13|
|ubuntu2004|cu118|3.8|2.0.1|1.18.0|0.14.2|1.17.0|0.16.13|
|ubuntu2004|cu118|3.10|2.0.1|1.18.0|0.14.2|1.17.0|0.16.13|
|ubuntu2004|cu118|3.8|2.2.2|1.18.0|0.14.2|1.17.0|0.16.13|
|ubuntu2004|cu118|3.10|2.2.2|1.18.0|0.14.2|1.17.0|0.16.13|
|ubuntu2004|cu121|3.8|2.2.2|1.18.0|0.14.2|1.17.0|0.16.13|
|ubuntu2004|cu121|3.10|2.2.2|1.18.0|0.14.2|1.17.0|0.16.13|
|ubuntu2004|cu118|3.10|2.3.0|1.18.0|0.14.2|1.17.0|0.16.13|
|ubuntu2004|cu121|3.10|2.3.0|1.18.0|0.14.2|1.17.0|0.16.13|
|ubuntu2004|cu121|3.8|2.1.2|1.18.0|0.14.2|1.17.0|0.16.13|
The following configurations are supported in the Microsoft Container Registry (MCR): [ptca_image_list.md](./ptca_image_list.md).

Other packages like fairscale, horovod, msccl, protobuf, pyspark, pytest, pytorch-lightning, tensorboard, NebulaML, torchvision, and torchmetrics are provided to support all training needs.

Expand Down
28 changes: 14 additions & 14 deletions docs/ecosystem/ptca_image_list.md
View file
Original file line number Diff line number Diff line change
@@ -1,14 +1,14 @@
| OS | GPU Type | Python Version | PyTorch Version | ORT-training version | DeepSpeed version | torch-ort Version | Nebula Version | Image Name | MCR Image Name |
| - | - | - | - | - | - | - | - | - | - |
|ubuntu2004|cu117|3.8|1.13.1|1.18.0|0.14.2|1.17.0|0.16.13|ptebic.azurecr.io/public/aifx/acpt/stable-ubuntu2004-cu117-py38-torch1131|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|ptebic.azurecr.io/public/aifx/acpt/stable-ubuntu2004-cu117-py39-torch1131|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|ptebic.azurecr.io/public/aifx/acpt/stable-ubuntu2004-cu117-py310-torch1131|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|ptebic.azurecr.io/public/aifx/acpt/stable-ubuntu2004-cu118-py38-torch201|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|ptebic.azurecr.io/public/aifx/acpt/stable-ubuntu2004-cu118-py310-torch201|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|ptebic.azurecr.io/public/aifx/acpt/stable-ubuntu2004-cu118-py38-torch222|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|ptebic.azurecr.io/public/aifx/acpt/stable-ubuntu2004-cu118-py310-torch222|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|ptebic.azurecr.io/public/aifx/acpt/stable-ubuntu2004-cu121-py38-torch222|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|ptebic.azurecr.io/public/aifx/acpt/stable-ubuntu2004-cu121-py310-torch222|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|ptebic.azurecr.io/public/aifx/acpt/stable-ubuntu2004-cu118-py310-torch230|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|ptebic.azurecr.io/public/aifx/acpt/stable-ubuntu2004-cu121-py310-torch230|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|ptebic.azurecr.io/public/aifx/acpt/stable-ubuntu2004-cu121-py38-torch212|mcr.microsoft.com/aifx/acpt/stable-ubuntu2004-cu121-py38-torch212|
| 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|

0 comments on commit ef5408e

Please sign in to comment.