Skip to content
This repository was archived by the owner on Dec 9, 2024. It is now read-only.

Commit c0cc464

Browse files
authored
Update Dockerfile_ubuntu_1804_tf_cuda_11_rollback
1 parent 484c0ce commit c0cc464

File tree

1 file changed

+2
-2
lines changed

1 file changed

+2
-2
lines changed

perfzero/docker/Dockerfile_ubuntu_1804_tf_cuda_11_rollback

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -95,11 +95,11 @@ RUN ${PIP_CMD} install -U scikit-learn
9595
# Install dependnecies needed for tf.distribute test utils
9696
RUN ${PIP_CMD} install dill tblib portpicker
9797

98-
RUN curl https://raw.githubusercontent.com/tensorflow/models/master/official/requirements.txt > /tmp/requirements.txt
98+
RUN curl curl https://raw.githubusercontent.com/tensorflow/benchmarks/master/perfzero/dockertest/requirements_temp.txt > /tmp/requirements.txt
9999
RUN ${PIP_CMD} install -r /tmp/requirements.txt
100100

101101
RUN ${PIP_CMD} install tf-estimator-nightly
102-
RUN ${PIP_CMD} install tensorflow-text-nightly==2.7.0.dev20210922
102+
RUN ${PIP_CMD} install tensorflow-text-nightly
103103

104104
# RUN nvidia-smi
105105

0 commit comments

Comments
 (0)