Do not install tensorflow 2.18.0 as it gives an error

images/tensorflow-notebook/cuda/Dockerfile

@@ -14,7 +14,7 @@ SHELL ["/bin/bash", "-o", "pipefail", "-c"]
 # Install TensorFlow, CUDA and cuDNN with pip
 RUN pip install --no-cache-dir \
     "jupyter-server-proxy" \
-    "tensorflow[and-cuda]" && \
+    "tensorflow[and-cuda]<=2.17.1" && \
     fix-permissions "${CONDA_DIR}" && \
     fix-permissions "/home/${NB_USER}"
 

tests/tensorflow-notebook/units/unit_tensorflow.py

@@ -1,13 +1,6 @@
 # Copyright (c) Jupyter Development Team.
 # Distributed under the terms of the Modified BSD License.
-import os
-
 import tensorflow as tf
 
-# Suppress logging warnings
-# https://stackoverflow.com/a/78803598
-os.environ["GRPC_VERBOSITY"] = "ERROR"
-os.environ["GLOG_minloglevel"] = "2"
-
 print(tf.constant("Hello, TensorFlow"))
 print(tf.reduce_sum(tf.random.normal([1000, 1000])))