diff --git a/dockerfile b/dockerfile
index c68809b..da38111 100644
--- a/dockerfile
+++ b/dockerfile
@@ -1,4 +1,4 @@
-# CUDA + cuDNN userspace from NVIDIA (Ubuntu 22.04)
+# CUDA 12.6 + cuDNN devel (Ubuntu 22.04)
 FROM nvidia/cuda:12.6.2-cudnn-devel-ubuntu22.04
 
 ENV DEBIAN_FRONTEND=noninteractive \
@@ -23,6 +23,8 @@ RUN apt-get update && apt-get install -y --no-install-recommends \
 RUN update-alternatives --install /usr/bin/python python /usr/bin/python3.10 1 \
  && update-alternatives --install /usr/bin/pip pip /usr/bin/pip3 1
 
+# ---- No cuDNN repo meddling needed if using TF 2.17.x ----
+
 # Python deps
 COPY requirements.txt /tmp/requirements.txt
 RUN pip install --upgrade pip \
@@ -40,7 +42,6 @@ RUN chmod +x /usr/local/bin/startup.sh
 
 EXPOSE 8888
 
-# Launch JupyterLab (tokenless for local dev; set a token if you want auth)
 CMD ["/bin/bash", "-lc", "/usr/local/bin/startup.sh && \
      exec jupyter lab --ip=0.0.0.0 --port=8888 --no-browser --allow-root \
      --ServerApp.token='' --ServerApp.password='' --ServerApp.root_dir=/data"]
\ No newline at end of file
diff --git a/requirements.txt b/requirements.txt
index 91cfdad..55c7f3d 100644
--- a/requirements.txt
+++ b/requirements.txt
@@ -1,11 +1,14 @@
 # --- Core training (Microwakeword) ---
 torch==2.5.1
 torchaudio==2.5.1
-tensorboard==2.18.0
+
+# TensorFlow stack (match CUDA/cuDNN in the base image)
+tensorflow==2.17.1
+tensorboard==2.17.1
 tensorboard-data-server==0.7.2
-tensorflow==2.18.0
-tensorflow-estimator==2.13.0
 tensorflow-io-gcs-filesystem==0.37.1
+# tensorflow-estimator not needed (remove)
+
 numpy==1.26.4
 scipy==1.12.0
 librosa==0.10.2.post1