diff --git a/Dockerfile b/Dockerfile
index b8d0000..c6c1ae7 100644
--- a/Dockerfile
+++ b/Dockerfile
@@ -1,4 +1,4 @@
-FROM ghcr.io/ucsd-ets/scipy-ml-notebook:stable
+FROM ghcr.io/ucsd-ets/datascience:stable
 
 LABEL maintainer="UC San Diego ITS/ETS <ets-consult@ucsd.edu>"
 
@@ -6,6 +6,25 @@ LABEL maintainer="UC San Diego ITS/ETS <ets-consult@ucsd.edu>"
 USER root
 
 
+RUN apt-get -q update && \
+  apt-get -qy install apt-utils  && \
+  apt-get -qy dist-upgrade && \
+  apt-get -qy auto-remove && \
+  apt-get install -qy p7zip-full software-properties-common && \
+  apt-get clean && \
+  rm -rf /var/lib/apt/lists/*  && \
+  wget -nv https://developer.download.nvidia.com/compute/cuda/repos/ubuntu2004/x86_64/cuda-ubuntu2004.pin -O /etc/apt/preferences.d/cuda-repository-pin-600 && \
+  apt-key adv --fetch-keys https://developer.download.nvidia.com/compute/cuda/repos/ubuntu2004/x86_64/3bf863cc.pub  && \
+  add-apt-repository \"deb http://developer.download.nvidia.com/compute/cuda/repos/ubuntu2004/x86_64/ /\"  && \
+  apt-get -q update && \
+  apt-get install -qqy cuda-11-1 cuda-nvcc-11-1	cuda-toolkit-11-1 && \
+  wget -nv https://developer.download.nvidia.com/compute/cuda/repos/ubuntu2004/x86_64/libcudnn8_8.0.5.39-1+cuda11.1_amd64.deb -O /var/tmp/libcudnn8.deb  && \
+  wget -nv https://developer.download.nvidia.com/compute/cuda/repos/ubuntu2004/x86_64/libnccl2_2.8.4-1+cuda11.1_amd64.deb -O /var/tmp/libnccl2.deb  && \
+  dpkg -i /var/tmp/libcudnn8.deb /var/tmp/libnccl2.deb  && \
+  fix-permissions $CONDA_DIR  && \
+  fix-permissions /home/$NB_USER
+
+
 # apt-get -q update
 # apt-get -qy install apt-utils
 # apt-get -qy dist-upgrade 
@@ -27,6 +46,7 @@ USER root
 #(nop)  ENV PATH=/opt/conda/bin:/usr/local/sbin:/usr/local/bin:/usr/sbin:/usr/bin:/sbin:/bin:/usr/local/cuda/bin
 
 
+COPY env.yaml /tmp/env.yml
 
 RUN conda env create --file /tmp/env.yml && \
     eval "$(conda shell.bash hook)" && \