diff --git a/Dockerfile b/Dockerfile
index d991caa..8ed602d 100644
--- a/Dockerfile
+++ b/Dockerfile
@@ -1,4 +1,4 @@
-FROM nvidia/cuda:12.6.1-cudnn-devel-ubuntu20.04
+FROM nvidia/cuda:12.4.1-cudnn-devel-ubuntu20.04
 
 WORKDIR /jup
 
@@ -18,13 +18,15 @@ RUN apt install -y software-properties-common \
 RUN apt update \
     && apt install -y python3.10 python3.10-venv python3.10-dev \
     libgl1-mesa-glx libffi-dev libssl-dev \
-    curl git htop vim nano\
-    openssh-server \
-    && curl https://bootstrap.pypa.io/get-pip.py -o get-pip.py \
+    curl git htop vim nano openssh-server
+
+RUN curl https://bootstrap.pypa.io/get-pip.py -o get-pip.py \
     && python3.10 get-pip.py \
-    && rm get-pip.py \
-    && /usr/bin/pip3 install -r requirements.txt -U \
-    && jupyter notebook --generate-config \
+    && rm get-pip.py
+
+RUN pip install -r requirements.txt -U
+
+RUN jupyter notebook --generate-config \
     && echo "c.FileCheckpoints.checkpoint_dir = '/tmp/'" >> /root/.jupyter/jupyter_notebook_config.py
 
 # setup sshd
@@ -32,8 +34,8 @@ RUN mkdir /var/run/sshd \
     && echo 'root:root' | chpasswd \
     && sed -i 's/#*PermitRootLogin prohibit-password/PermitRootLogin yes/g' /etc/ssh/sshd_config \
     && sed -i 's@session\s*required\s*pam_loginuid.so@session optional pam_loginuid.so@g' /etc/pam.d/sshd \
-    && echo "export VISIBLE=now" >> /etc/profile 
+    && echo "export VISIBLE=now" >> /etc/profile
 
 EXPOSE 8888 22
 
-ENTRYPOINT ["jupyter", "lab","--ip=0.0.0.0","--allow-root","--NotebookApp.token=''","--NotebookApp.password=''"]
+ENTRYPOINT ["jupyter", "lab","--ip=0.0.0.0","--allow-root","--NotebookApp.token=''","--NotebookApp.password=''"]
\ No newline at end of file