diff --git a/Dockerfile b/Dockerfile
index 57ca661..5fff46f 100644
--- a/Dockerfile
+++ b/Dockerfile
@@ -1,4 +1,4 @@
-FROM ghcr.io/ucsd-ets/datascience-notebook:2023.4-stable
+FROM ghcr.io/ucsd-ets/datascience-notebook:2024.4-stable
 
 LABEL maintainer="UC San Diego ITS/ETS <ets-consult@ucsd.edu>"
 #
@@ -31,7 +31,7 @@ RUN     apt-get -q update && \
 
 COPY env.yml /tmp/env.yml
 
-RUN conda env create --file /tmp/env.yml && \
+RUN mamba env create --file /tmp/env.yml && \
     eval "$(conda shell.bash hook)" && \
     conda activate ${KERNEL} && \
     mkdir -p $CONDA_PREFIX/etc/conda/activate.d && \
@@ -47,7 +47,6 @@ RUN conda env create --file /tmp/env.yml && \
 USER jovyan
 # other packages here... 
 
-
 # From the "original" 2022 version: 
 # USER    $NB_UID:$NB_GID
 ENV     PATH=${PATH}:/usr/local/cuda/bin
diff --git a/env.yml b/env.yml
index b2af90d..c202053 100644
--- a/env.yml
+++ b/env.yml
@@ -1,37 +1,31 @@
 name: vis149
 channels:
-- conda-forge
-- pytorch-nightly
-- pytorch
-- defaults
+  - conda-forge
+  - pytorch
+  - defaults
 dependencies:
-- python=3.7
-#- python=3.9
-#- tensorflow=2.9.1=cuda112*
-- cudatoolkit=11.1
-- cudnn=8.0.5
-- jaxlib=0.1.55
-#- tensorboard
-- ipykernel
-- matplotlib
-- pip
-- pip:
-  - click
-  - requests
-  - tqdm
-  - gdown
-  - imageio-ffmpeg==0.4.3
-  - jax==0.1.73
-  - opencv-contrib-python-headless
-  - opencv-python
-  - opensimplex
-  - pillow
-  - pyspng==0.1.0
-  - networkx
-  - scipy
-  - ninja
-  - torch==1.8.0+cu111
-  - torchvision==0.9.0+cu111
-  - torchaudio==0.8.0
-#  - tensorflow==2.12.*
-#  - nvidia-cudnn-cu11==8.6.0.163
+  - python=3.7
+  - cudatoolkit=11.1
+  - cudnn=8.1.0
+  - jaxlib=0.1.55
+  - pip
+  - pip:
+      - ipykernel
+      - matplotlib
+      - click
+      - requests
+      - tqdm
+      - gdown
+      - jax==0.1.73
+      - imageio-ffmpeg==0.4.3
+      - opencv-contrib-python-headless
+      - opencv-python
+      - opensimplex
+      - pillow
+      - pyspng==0.1.0
+      - networkx
+      - scipy
+      - ninja
+      - torch==1.8.0
+      - torchvision==0.9.0
+      - torchaudio==0.8.0
\ No newline at end of file