File tree 1 file changed +1
-1
lines changed
tensorflow/training/docker/2.3/py3/cu110 1 file changed +1
-1
lines changed Original file line number Diff line number Diff line change @@ -263,7 +263,7 @@ RUN cat /etc/ssh/ssh_config | grep -v StrictHostKeyChecking > /etc/ssh/ssh_confi
263
263
ENV TF_AUTOTUNE_THRESHOLD=2
264
264
265
265
# 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
267
267
268
268
# Install SM Distributed DataParallel binary
269
269
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
You can’t perform that action at this time.
0 commit comments