From 742aad523478ab2aa484ae43fff56d0e31a7aed1 Mon Sep 17 00:00:00 2001 From: Sophie Schoenmeyer <107952697+sophies927@users.noreply.github.com> Date: Wed, 29 May 2024 21:11:46 +0000 Subject: [PATCH] Add updated list of images --- docs/ecosystem/acpt.md | 24 +++++++++++++++--------- docs/ecosystem/ptca_image_list.md | 14 ++++++++++++++ 2 files changed, 29 insertions(+), 9 deletions(-) create mode 100644 docs/ecosystem/ptca_image_list.md diff --git a/docs/ecosystem/acpt.md b/docs/ecosystem/acpt.md index e123add64fda9..3b79c2990b9a7 100644 --- a/docs/ecosystem/acpt.md +++ b/docs/ecosystem/acpt.md @@ -1,7 +1,7 @@ --- title: Azure Container for PyTorch (ACPT) description: Learn more about Azure Container for PyTorch (ACPT) and how it utilizes ONNX Runtime -nav_order: 9 +nav_order: 1 redirect_from: /docs/tutorials/ecosystem/acpt --- # Azure Container for PyTorch (ACPT) @@ -27,14 +27,20 @@ Azure Container for PyTorch (ACPT) is a lightweight, standalone environment that ## Supported configurations for Azure Container for PyTorch (ACPT) The following configurations are supported in the Microsoft Container Registry (MCR): -Most recent image releases (last updated 4/25/2024): - -| Environment Name | OS | GPU Version | Python Version | PyTorch Version | ORT-training Version | DeepSpeed Version | torch-ort Version | Nebula Version | -|-----------------------------|----------------|-------------|----------------|-----------------|----------------------|-------------------|-------------------|----------------| -| acpt-pytorch-2.2-cuda12.1 | Ubuntu 20.04 | cu121 | 3.10 | 2.2.2 | 1.18.0 | 0.14.2 | 1.17.3 | 0.16.11 | -| acpt-pytorch-2.1-cuda12.1 | Ubuntu 20.04 | cu121 | 3.10 | 2.1.2 | 1.18.0 | 0.14.2 | 1.17.3 | 0.16.11 | -| acpt-pytorch-2.0-cuda11.7 | Ubuntu 20.04 | cu117 | 3.10 | 2.0.1 | 1.18.0 | 0.14.2 | 1.17.3 | 0.16.11 | -| acpt-pytorch-1.13-cuda11.7 | Ubuntu 20.04 | cu117 | 3.10 | 1.13.1 | 1.18.0 | 0.14.2 | 1.17.3 | 0.16.11 | +| 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| Other packages like fairscale, horovod, msccl, protobuf, pyspark, pytest, pytorch-lightning, tensorboard, NebulaML, torchvision, and torchmetrics are provided to support all training needs. diff --git a/docs/ecosystem/ptca_image_list.md b/docs/ecosystem/ptca_image_list.md new file mode 100644 index 0000000000000..9b1f8dc702afa --- /dev/null +++ b/docs/ecosystem/ptca_image_list.md @@ -0,0 +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|