Skip to content

Commit 668e65d

Browse files
authored
fix: Add PyTorch 2.1.0 SM Training DLC to UNSUPPORTED_DLC_IMAGE_FOR_SM_PARALLELISM list (aws#4356)
* add 2.1 unsupported smddp * formatting
1 parent d083396 commit 668e65d

File tree

1 file changed

+5
-1
lines changed

1 file changed

+5
-1
lines changed

src/sagemaker/estimator.py

+5-1
Original file line numberDiff line numberDiff line change
@@ -3282,7 +3282,11 @@ class Framework(EstimatorBase):
32823282
"""
32833283

32843284
_framework_name = None
3285-
UNSUPPORTED_DLC_IMAGE_FOR_SM_PARALLELISM = ("2.0.1-gpu-py310-cu121", "2.0-gpu-py310-cu121")
3285+
UNSUPPORTED_DLC_IMAGE_FOR_SM_PARALLELISM = (
3286+
"2.0.1-gpu-py310-cu121",
3287+
"2.0-gpu-py310-cu121",
3288+
"2.1.0-gpu-py310",
3289+
)
32863290

32873291
def __init__(
32883292
self,

0 commit comments

Comments
 (0)