tensorflow data validation

Author: hephaex

python-tensorflow/Dockerfile.2.8-py38-cuda11.3

@@ -1,4 +1,4 @@
-FROM lablup/common-base:py38-cuda11.2
+FROM lablup/common-base:py38-cuda11.3
 
 RUN python3 -m pip install --no-cache-dir --upgrade \
 	    -q git+https://github.com/google-research/tensorflow_constrained_optimization
@@ -38,20 +38,22 @@ RUN python3 -m pip install --no-cache-dir \
 	    mesh-tensorflow==0.1.18 \
             tensorflow-cloud==0.1.10 \
             tensorflow-hub==0.9.0 
-
-RUN python3 -m pip install --no-cache-dir \
-	    -i https://pypi-nightly.tensorflow.org/simple tensorflow-data-validation
+RUN python3 -m pip uninstall -y apache-beam[gcp] && \
+    python3 -m pip install --no-cache-dir  apache-beam[gcp]==2.38.0
+RUN   python3 -m pip install --no-cache-dir \
+            tensorflow-gpu==2.8.0 \
+	    tensorboard-plugin-wit \
+	    tensorboard numpy scipy
 RUN python3 -m pip install --no-cache-dir \
 	    tensorflow-graphics-gpu==1.0.0 \
-            tensorflow-io==0.17.0 \
+            tensorflow-io==0.26.0 \
 	    tensorflow_ranking==0.3.2 \
 	    tensorflow-plot==0.3.2 \
-	    tensorflow_text==2.4.1 \
+	    tensorflow_text==2.8.2 \
 	    tensorflow-gan==2.0.0 
-RUN   python3 -m pip install --no-cache-dir \
-            tensorflow-gpu==2.8.0 \
-	    tensorboard-plugin-wit \
-	    tensorboard numpy scipy
+RUN python3 -m pip install --no-cache-dir pyfarmhash==0.3.2 pyarrow==8.0.0 && \
+    python3 -m pip install --no-cache-dir \
+	    -i https://pypi-nightly.tensorflow.org/simple tensorflow-data-validation==1.8.0.dev20220515
 
 # Install Horovod, temporarily using CUDA stubs
 RUN ldconfig /usr/local/cuda/targets/x86_64-linux/lib/stubs && \