From a3ea386b7c34145107926d469cd3a10c0d21e3c5 Mon Sep 17 00:00:00 2001
From: Erik Stambaugh <erik@erikstambaugh.com>
Date: Thu, 22 Feb 2024 06:37:39 -0800
Subject: [PATCH] Getting closer to having TF working with CUDA

---
 jupyter/Dockerfile | 30 ++++++++++++------------------
 1 file changed, 12 insertions(+), 18 deletions(-)

diff --git a/jupyter/Dockerfile b/jupyter/Dockerfile
index 35fbe78..8b494df 100644
--- a/jupyter/Dockerfile
+++ b/jupyter/Dockerfile
@@ -1,34 +1,28 @@
 
 FROM continuumio/miniconda3
 
-RUN    conda config --quiet --add channels conda-forge
-RUN    conda config --quiet --add channels nvidia
+RUN conda update -y -n base -c conda-forge conda
 
-RUN conda update conda -y
-
-RUN    conda install --quiet -y -c nvidia \
-         cudatoolkit=11.8 \
-         cudnn
-
-RUN conda install --quiet -y \
-        python=3.9
+RUN conda create -y --name jupyter python=3.10
 
 RUN CONDA_OVERRIDE_CUDA=11.8 \
-    conda install --quiet -y -c conda-forge \
+    conda install -n jupyter --quiet -y -c conda-forge \
          jupyterlab \
          pandas \
          librosa \
          matplotlib \
          pyarrow \
          pillow \
-         tensorflow-gpu=2.14 \
          keras \
+         cudatoolkit=11.8 \
+         cudnn \
+         keras-tuner \
+         numpy \
+         h5py \
+         tensorflow=2.14 \
     && /bin/true  # only added to make the installed package lines consistent
 
-CMD /opt/conda/bin/jupyter lab \
-        --notebook-dir=/notebooks \
-        --ip='*' \
-        --port=9001 \
-        --no-browser \
-        --allow-root
+SHELL      ["conda",   "run",                        "-n", "jupyter", "/bin/bash",  "-c"]
+ENTRYPOINT ["conda",   "run", "--no-capture-output", "-n", "jupyter" ]
+CMD        ["jupyter", "lab", "--ip", "0.0.0.0", "--port", "9001", "--no-browser", "--allow-root", "--LabApp.token=''", "--notebook-dir=/notebooks", "--ResourceUseDisplay.track_cpu_percent=True" ]