diff --git a/Makefile b/Makefile
index aa40a3b..5cd969e 100644
--- a/Makefile
+++ b/Makefile
@@ -7,7 +7,7 @@ up: run
 run: build
 	docker compose up
 
-build: docker-compose.yaml jupyter/Dockerfile
+build: jupyter/Dockerfile
 	docker compose build
 
 down:
diff --git a/docker-compose.yaml b/docker-compose.yaml
index f0f60cd..acd61f6 100644
--- a/docker-compose.yaml
+++ b/docker-compose.yaml
@@ -4,13 +4,13 @@ services:
   jupyter:
     container_name: jupyter
     build:
-      context: .
-      dockerfile: jupyter/Dockerfile
+      context: ./jupyter
+      dockerfile: Dockerfile
     restart: unless-stopped
     ports:
       - 0.0.0.0:9001:9001
     volumes:
-      - ./jupyter/notebooks:/notebooks
+      - ./notebooks:/notebooks
       - ./jupyter/overrides.json:/opt/conda/share/jupyter/lab/settings/overrides.json
     environment:
       - JUPYTER_TOKEN=12345
diff --git a/jupyter/Dockerfile b/jupyter/Dockerfile
index de71f15..35fbe78 100644
--- a/jupyter/Dockerfile
+++ b/jupyter/Dockerfile
@@ -1,30 +1,30 @@
 
 FROM continuumio/miniconda3
 
-# docker run -i -t -p 8888:8888 continuumio/miniconda3 /bin/bash \
-# -c "/opt/conda/bin/conda install jupyter -y --quiet && mkdir \
-# /opt/notebooks && /opt/conda/bin/jupyter notebook \
-# --notebook-dir=/opt/notebooks --ip='*' --port=8888 \
-# --no-browser --allow-root"
+RUN    conda config --quiet --add channels conda-forge
+RUN    conda config --quiet --add channels nvidia
 
+RUN conda update conda -y
 
-RUN    conda config --quiet --add channels conda-forge \
-    && conda install -c conda-forge --quiet -y jupyterlab
+RUN    conda install --quiet -y -c nvidia \
+         cudatoolkit=11.8 \
+         cudnn
 
-#RUN    conda install -c conda-forge --quiet --freeze-installed -y \
-RUN    conda install -c conda-forge --quiet -y \
+RUN conda install --quiet -y \
+        python=3.9
+
+RUN CONDA_OVERRIDE_CUDA=11.8 \
+    conda install --quiet -y -c conda-forge \
+         jupyterlab \
          pandas \
          librosa \
          matplotlib \
          pyarrow \
-         pillow=9.4.0 \
-         tensorflow \
+         pillow \
+         tensorflow-gpu=2.14 \
          keras \
     && /bin/true  # only added to make the installed package lines consistent
 
-# (NO) libtiff
-# pillow
-
 CMD /opt/conda/bin/jupyter lab \
         --notebook-dir=/notebooks \
         --ip='*' \