Merge pull request #751 from NVIDIA/gh/release

[WideAndDeep/TF] library version fix
NVIDIA · Nov 10, 2020 · 3ddcba4 · 3ddcba4
2 parents 7e85343 + fa1ddc9
commit 3ddcba4
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/TensorFlow/Recommendation/WideAndDeep/Dockerfile b/TensorFlow/Recommendation/WideAndDeep/Dockerfile
@@ -40,7 +40,7 @@ ENV PYTHONPATH $SPARK_HOME/python:$SPARK_HOME/python/lib/py4j-0.10.7-src.zip
 ENV SPARK_OPTS --driver-java-options=-Xms1024M --driver-java-options=-Xmx4096M --driver-java-options=-Dlog4j.logLevel=info
 
 RUN pip install pyspark==2.3.1
-RUN pip install --no-deps tensorflow-transform apache-beam==2.14 tensorflow-metadata==0.14.0 pydot dill
+RUN pip install --no-deps tensorflow-transform==0.24.1 apache-beam==2.14 tensorflow-metadata==0.14.0 pydot dill
 RUN pip install ipdb
 RUN pip install -e git://github.com/NVIDIA/dllogger#egg=dllogger