Added p5 as a supported NCCL instance

aws · Sep 5, 2024 · b6df651 · b6df651
1 parent 628166c
commit b6df651
Showing 1 changed file with 1 addition and 0 deletions.
diff --git a/src/sagemaker_training/__init__.py b/src/sagemaker_training/__init__.py
@@ -81,6 +81,7 @@
     "ml.p3dn.24xlarge",
     "ml.p4d.24xlarge",
     "ml.p4de.24xlarge",
+    "ml.p5.48xlarge",
     "ml.trn1.32xlarge",
 ]