Skip to content

Commit 71d4892

Browse files
authored
fix package_version_regression (#5159)
1 parent d7769ff commit 71d4892

File tree

2 files changed

+6
-0
lines changed

2 files changed

+6
-0
lines changed

pytorch/training/docker/2.7/py3/Dockerfile.sagemaker.cpu.core_packages.json

Lines changed: 3 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -64,5 +64,8 @@
6464
},
6565
"opencv-python": {
6666
"version_specifier": "==4.11.0.86"
67+
},
68+
"sagemaker-pytorch-training": {
69+
"version_specifier": ">=2.9.0,<3"
6770
}
6871
}

pytorch/training/docker/2.7/py3/cu128/Dockerfile.sagemaker.gpu.core_packages.json

Lines changed: 3 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -71,5 +71,8 @@
7171
},
7272
"awscli": {
7373
"version_specifier": "<2"
74+
},
75+
"sagemaker-pytorch-training": {
76+
"version_specifier": ">=2.9.0,<3"
7477
}
7578
}

0 commit comments

Comments
 (0)