File tree Expand file tree Collapse file tree 3 files changed +9
-9
lines changed
pytorch/training/docker/2.7/py3 Expand file tree Collapse file tree 3 files changed +9
-9
lines changed Original file line number Diff line number Diff line change @@ -124,7 +124,7 @@ nightly_pr_test_mode = false
124124dlc-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"
128128dlc-pr-tensorflow-2-training = " "
129129dlc-pr-autogluon-training = " "
130130
Original file line number Diff line number Diff 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
334334RUN 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" \
Original file line number Diff line number Diff 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
205205RUN 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" \
You can’t perform that action at this time.
0 commit comments