Skip to content

Commit fe96507

Browse files
committed
unpin pt 2.7 sm
1 parent 77ab1db commit fe96507

File tree

3 files changed

+9
-9
lines changed

3 files changed

+9
-9
lines changed

dlc_developer_config.toml

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -124,7 +124,7 @@ nightly_pr_test_mode = false
124124
dlc-pr-base = ""
125125

126126
# Standard Framework Training
127-
dlc-pr-pytorch-training = "pytorch/training/buildspec-2-8-ec2.yml"
127+
dlc-pr-pytorch-training = "pytorch/training/buildspec-2-7-sm.yml"
128128
dlc-pr-tensorflow-2-training = ""
129129
dlc-pr-autogluon-training = ""
130130

pytorch/training/docker/2.7/py3/Dockerfile.cpu

Lines changed: 4 additions & 4 deletions
Original file line numberDiff line numberDiff line change
@@ -251,7 +251,7 @@ RUN pip install --no-cache-dir -U torch==${PYTORCH_VERSION} \
251251
#thinc 8.3.6 is not compatible with numpy 1.26.4 (sagemaker doesn't support latest numpy)
252252
thinc==8.3.4 \
253253
blis \
254-
typer \
254+
typer \
255255
langcodes \
256256
language_data \
257257
numpy \
@@ -325,16 +325,16 @@ RUN pip install --no-cache-dir -U torch==${PYTORCH_VERSION} \
325325
thinc==8.3.4 \
326326
blis \
327327
numpy \
328-
typer \
328+
typer \
329329
langcodes \
330330
language_data \
331331
&& pip uninstall -y dataclasses
332332

333333
# Install SM packages
334334
RUN pip install --no-cache-dir -U \
335335
# address package regression caused by smclarify depedency s3fs"
336-
"awscli<=1.42.61" \
337-
"boto3<=1.40.61" \
336+
awscli \
337+
boto3 \
338338
smclarify \
339339
"sagemaker>=2.9.0,<3" \
340340
"sagemaker-experiments<1" \

pytorch/training/docker/2.7/py3/cu128/Dockerfile.gpu

Lines changed: 4 additions & 4 deletions
Original file line numberDiff line numberDiff line change
@@ -71,7 +71,7 @@ RUN pip install --no-cache-dir -U torch==${PYTORCH_VERSION} \
7171
--index-url https://download.pytorch.org/whl/cu128 \
7272
&& pip install --no-cache-dir -U torchtnt==${TORCHTNT_VERSION} \
7373
torchdata==${TORCHDATA_VERSION} \
74-
triton
74+
triton
7575

7676
# Install GDRCopy which is a dependency of SM Distributed DataParallel binary
7777
# The test binaries requires cuda driver library which could be found in conda
@@ -152,7 +152,7 @@ RUN pip install --no-cache-dir \
152152
#thinc 8.3.6 is not compatible with numpy 1.26.4 (sagemaker doesn't support latest numpy)
153153
"thinc==8.3.4" \
154154
blis \
155-
typer \
155+
typer \
156156
langcodes \
157157
language_data \
158158
"jinja2>=3.1.6"\
@@ -204,8 +204,8 @@ RUN chmod +x /usr/local/bin/start_with_right_hostname.sh
204204
# Install SM packages
205205
RUN pip install --no-cache-dir -U \
206206
# address package regression caused by smclarify depedency s3fs"
207-
"awscli<=1.42.61" \
208-
"boto3<=1.40.61" \
207+
awscli \
208+
boto3 \
209209
smclarify \
210210
"sagemaker>=2.9.0,<3" \
211211
"sagemaker-experiments<1" \

0 commit comments

Comments
 (0)