update tensorflow-probability

Author: hephaex

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

@@ -14,13 +14,13 @@ RUN python3 -m pip install --no-cache-dir \
 	    tensorflow-transform==1.10.1 \
 	    tensorflow_model_analysis==0.41.1 \
 	    tensorflow-datasets==4.7.0 \
-	    tensorflow_probability==0.11.1 \ 
+	    tensorflow_probability==0.18.0 \ 
 	    tensorflow-addons==0.11.2 \
 	    neural-structured-learning==1.3.1 \
 	    tensorflow-model-optimization==0.5.0 \
 	    tfx==1.10.0 \
 	    tfx-bsl==1.10.1 \ 
-	    tf-agents==0.6.0  
+	    tf-agents==0.14.0  
 
 RUN python3 -m pip install --no-cache-dir \
       --use-feature=2020-resolver \
@@ -49,7 +49,7 @@ RUN python3 -m pip install --no-cache-dir \
 	    tensorflow_text==2.4.1 \
 	    tensorflow-gan==2.0.0 
 RUN   python3 -m pip install --no-cache-dir \
-            tensorflow-gpu==2.8.0 \
+            tensorflow-gpu==2.10.0 \
 	    tensorboard-plugin-wit \
 	    tensorboard numpy scipy