Skip to content

Commit db6cc3b

Browse files
pin all sm packages and rebuild
1 parent 2bea98d commit db6cc3b

File tree

2 files changed

+11
-12
lines changed

2 files changed

+11
-12
lines changed

tensorflow/training/docker/2.19/py3/Dockerfile.cpu

Lines changed: 6 additions & 6 deletions
Original file line numberDiff line numberDiff line change
@@ -287,18 +287,18 @@ RUN $PYTHON -m pip install --no-cache-dir -U \
287287
plotly \
288288
seaborn \
289289
shap
290-
290+
291291
RUN $PYTHON -m pip install --no-cache-dir -U \
292-
sagemaker \
293-
sagemaker-experiments \
292+
sagemaker==2.254.1 \
293+
sagemaker-experiments==0.1.45 \
294294
sagemaker-tensorflow-training==20.4.1 \
295-
sagemaker-training \
295+
sagemaker-training==5.1.1 \
296296
y-py==0.6.2 \
297297
sagemaker-studio-analytics-extension==0.1.7 \
298298
sparkmagic==0.22.0 \
299299
sagemaker-studio-sparkmagic-lib==0.1.4 \
300-
smclarify
301-
300+
smclarify==0.5
301+
302302
# Remove python kernel installed by sparkmagic
303303
RUN /usr/local/bin/jupyter-kernelspec remove -f python3
304304

tensorflow/training/docker/2.19/py3/cu125/Dockerfile.gpu

Lines changed: 5 additions & 6 deletions
Original file line numberDiff line numberDiff line change
@@ -359,18 +359,17 @@ RUN $PYTHON -m pip install --no-cache-dir -U \
359359
plotly \
360360
seaborn \
361361
shap
362-
362+
363363
RUN $PYTHON -m pip install --no-cache-dir -U \
364-
sagemaker \
365-
sagemaker-experiments \
364+
sagemaker==2.254.1 \
365+
sagemaker-experiments==0.1.45 \
366366
sagemaker-tensorflow-training==20.4.1 \
367-
sagemaker-training \
367+
sagemaker-training==5.1.1 \
368368
y-py==0.6.2 \
369369
sagemaker-studio-analytics-extension==0.1.7 \
370370
sparkmagic==0.22.0 \
371371
sagemaker-studio-sparkmagic-lib==0.1.4 \
372-
smclarify
373-
372+
smclarify==0.5
374373
# install boost
375374
# tensorflow is compiled with --cxxopt="-D_GLIBCXX_USE_CXX11_ABI=1"
376375
RUN wget https://sourceforge.net/projects/boost/files/boost/1.82.0/boost_1_82_0.tar.gz/download -O boost_1_82_0.tar.gz \

0 commit comments

Comments
 (0)