Added p5 as a supported NCCL instance

andjsmi · 992X · commit b6df651cb52d · 2024-09-05T10:22:12.000-07:00
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",
 ]
 
