Skip to content

Commit 08d6629

Browse files
authored
Update resource-azure-container-for-pytorch.md
1 parent 9609eaa commit 08d6629

File tree

1 file changed

+4
-3
lines changed

1 file changed

+4
-3
lines changed

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

Lines changed: 4 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -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)