Skip to content

Commit 3ddcba4

Browse files
Merge pull request NVIDIA#751 from NVIDIA/gh/release
[WideAndDeep/TF] library version fix
2 parents 7e85343 + fa1ddc9 commit 3ddcba4

File tree

1 file changed

+1
-1
lines changed

1 file changed

+1
-1
lines changed

TensorFlow/Recommendation/WideAndDeep/Dockerfile

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -40,7 +40,7 @@ ENV PYTHONPATH $SPARK_HOME/python:$SPARK_HOME/python/lib/py4j-0.10.7-src.zip
4040
ENV SPARK_OPTS --driver-java-options=-Xms1024M --driver-java-options=-Xmx4096M --driver-java-options=-Dlog4j.logLevel=info
4141

4242
RUN pip install pyspark==2.3.1
43-
RUN pip install --no-deps tensorflow-transform apache-beam==2.14 tensorflow-metadata==0.14.0 pydot dill
43+
RUN pip install --no-deps tensorflow-transform==0.24.1 apache-beam==2.14 tensorflow-metadata==0.14.0 pydot dill
4444
RUN pip install ipdb
4545
RUN pip install -e git://github.com/NVIDIA/dllogger#egg=dllogger
4646

0 commit comments

Comments
 (0)