Skip to content

Commit 8727775

Browse files
Merge pull request #275292 from ssalgadodev/patch-103
Update resource-azure-container-for-pytorch.md
2 parents b1cafd3 + 1fdbb7c commit 8727775

File tree

1 file changed

+5
-4
lines changed

1 file changed

+5
-4
lines changed

articles/machine-learning/resource-azure-container-for-pytorch.md

Lines changed: 5 additions & 4 deletions
Original file line numberDiff line numberDiff line change
@@ -10,7 +10,7 @@ ms.service: machine-learning
1010
ms.subservice: core
1111
ms.custom: build-2023, build-2023-dataai
1212
ms.topic: reference
13-
ms.date: 03/20/2023
13+
ms.date: 05/13/2024
1414
---
1515

1616
# Azure Container for PyTorch (ACPT)
@@ -42,9 +42,10 @@ The following configurations are supported:
4242

4343
| Environment Name | OS | GPU Version| Python Version | PyTorch Version | ORT-training Version | DeepSpeed Version | torch-ort Version | Nebula Version |
4444
| --- | --- | --- | --- | --- | --- | --- | --- | --- |
45-
|acpt-pytorch-2.1-cuda12.1|Ubuntu 20.04|cu121|3.8|2.1.2|1.17.1|0.13.1|1.17.0|0.16.10|
46-
|acpt-pytorch-2.0-cuda11.7|Ubuntu 20.04|cu117|3.8|2.0.1|1.17.1|0.13.1|1.17.0|0.16.10|
47-
|acpt-pytorch-1.13-cuda11.7|Ubuntu 20.04|cu117|3.8|1.13.1|1.17.1|0.13.1|1.17.0|0.16.10|
45+
|acpt-pytorch-2.2-cuda12.1|Ubuntu 20.04|cu121|3.10|2.2.2|1.17.3|0.13.1|1.17.3|0.16.11|
46+
|acpt-pytorch-2.1-cuda12.1|Ubuntu 20.04|cu121|3.10|2.1.2|1.17.3|0.13.1|1.17.3|0.16.11|
47+
|acpt-pytorch-2.0-cuda11.7|Ubuntu 20.04|cu117|3.10|2.0.1|1.17.3|0.13.1|1.17.3|0.16.11|
48+
|acpt-pytorch-1.13-cuda11.7|Ubuntu 20.04|cu117|3.10|1.13.1|1.17.3|0.13.1|1.17.3|0.16.11|
4849

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

0 commit comments

Comments
 (0)