Skip to content

Commit fa1ddc9

Browse files
committed
[WideAndDeep/TF] library version fix
1 parent 7e85343 commit fa1ddc9

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)