Skip to content

Commit e528a56

Browse files
Update SMP TF 2.3.1 binary with bug fix (#840)
Co-authored-by: Tejas Chumbalkar <[email protected]>
1 parent 5d4537a commit e528a56

File tree

1 file changed

+1
-1
lines changed

1 file changed

+1
-1
lines changed

tensorflow/training/docker/2.3/py3/cu110/Dockerfile.gpu

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -263,7 +263,7 @@ RUN cat /etc/ssh/ssh_config | grep -v StrictHostKeyChecking > /etc/ssh/ssh_confi
263263
ENV TF_AUTOTUNE_THRESHOLD=2
264264

265265
# Install SMD MP binary
266-
RUN pip install --no-cache-dir -U https://sagemaker-distributed-model-parallel.s3.amazonaws.com/tensorflow-2.3/build-artifacts/2020-12-06/smdistributed_modelparallel-1.0.0-cp37-cp37m-linux_x86_64.whl
266+
RUN pip install --no-cache-dir -U https://sagemaker-distributed-model-parallel.s3.amazonaws.com/tensorflow-2.3/build-artifacts/2021-02-12-04-00/smdistributed_modelparallel-1.2.0-cp37-cp37m-linux_x86_64.whl
267267

268268
# Install SM Distributed DataParallel binary
269269
ARG SMDATAPARALLEL_BINARY=https://smdataparallel.s3.amazonaws.com/binary/tensorflow/2.3.1/cu110/2021-01-14/smdistributed_dataparallel-1.0.0-cp37-cp37m-linux_x86_64.whl

0 commit comments

Comments
 (0)