trying out a fix

Dockerfile

@@ -2,39 +2,32 @@ FROM nvcr.io/nvidia/tensorflow:20.12-tf1-py3
 
 USER root
 
-# Fix GPG Keys and install system libs
+# Fix GPG Keys
 RUN apt-key adv --fetch-keys https://developer.download.nvidia.com/compute/cuda/repos/ubuntu2004/x86_64/3bf863cc.pub \
     && apt-get update && apt-get install -y libgl1-mesa-glx libglib2.0-0 \
     && rm -rf /var/lib/apt/lists/*
 
-# FORCE Python 3.7 as the only 'python' and 'python3'
-# This stops the system from accidentally using 3.8
-RUN ln -sf /usr/bin/python3.7 /usr/bin/python3 && \
-    ln -sf /usr/bin/python3.7 /usr/bin/python && \
-    ln -sf /usr/bin/python3.7 /usr/local/bin/python
-
 # Set up user
 RUN useradd -m -u 1000 user
 RUN mkdir -p /home/user/app && chown -R user:user /home/user/app
 
-# Strict Environment Paths
+# Force the environment to use the NVIDIA Python 3.7 site-packages
 ENV HOME=/home/user \
     PATH=/home/user/.local/bin:/usr/local/bin:/usr/bin:/sbin:/bin \
-    PYTHONPATH=/usr/local/lib/python3.7/dist-packages:/home/user/app \
+    PYTHONPATH=/usr/local/lib/python3.7/dist-packages \
     HF_HOME=/home/user/app/.cache
 
 WORKDIR /home/user/app
 
-# Upgrade pip for 3.7 specifically
-RUN python -m pip install --no-cache-dir --upgrade "pip<23.0" "setuptools<60"
+# Use the absolute path to the binary provided by NVIDIA
+RUN /usr/bin/python3 -m pip install --no-cache-dir --upgrade "pip<23.0" "setuptools<60"
 
-# Install requirements
 COPY requirements.txt .
-RUN python -m pip install --no-cache-dir -r requirements.txt
+RUN /usr/bin/python3 -m pip install --no-cache-dir -r requirements.txt
 
 COPY --chown=user:user . .
 
 USER user
 
-# Using 'python -m uvicorn' ensures we use the version linked above
-CMD ["python", "-m", "uvicorn", "app:app", "--host", "0.0.0.0", "--port", "7860"]
+# Use the absolute path for the CMD to ensure no version drift
+CMD ["/usr/bin/python3", "-m", "uvicorn", "app:app", "--host", "0.0.0.0", "--port", "7860"]

app.py

@@ -6,10 +6,12 @@ import tensorflow as tf
 import sys 
 
 try:
-    import keras
-except ImportError:
+    import tensorflow.compat.v1 as tf
+    tf.disable_v2_behavior() # Force TF1 behavior
     from tensorflow import keras
-
+except ImportError:
+    import keras
+    
 import keras.engine as KE 
 sys.modules['keras.engine'] = keras.engine
 sys.modules['keras.layers'] = keras.layers