diff --git a/Dockerfile b/Dockerfile
index 5c62eaa..0eaa6b9 100644
--- a/Dockerfile
+++ b/Dockerfile
@@ -98,17 +98,24 @@
 # xserver-xorg-legacy && \
 #RUN sed -i "s/allowed_users=console/allowed_users=anybody/;$ a needs_root_rights=yes" /etc/X11/Xwrapper.config
 
+#RUN cd /tmp && \
+#    wget https://developer.download.nvidia.com/compute/cuda/repos/ubuntu1804/x86_64/cuda-ubuntu1804.pin && \
+#    mv cuda-ubuntu1804.pin /etc/apt/preferences.d/cuda-repository-pin-600 && \
+#    wget https://developer.download.nvidia.com/compute/cuda/11.6.0/local_installers/cuda-repo-ubuntu1804-11-6-local_11.6.0-510.39.01-1_amd64.deb && \
+#    dpkg -i cuda-repo-ubuntu1804-11-6-local_11.6.0-510.39.01-1_amd64.deb && \
+#    apt-key add /var/cuda-repo-ubuntu1804-11-6-local/7fa2af80.pub
+
 RUN cd /tmp && \
     wget https://developer.download.nvidia.com/compute/cuda/repos/ubuntu1804/x86_64/cuda-ubuntu1804.pin && \
-    mv cuda-ubuntu1804.pin /etc/apt/preferences.d/cuda-repository-pin-600 && \
-    wget https://developer.download.nvidia.com/compute/cuda/11.6.0/local_installers/cuda-repo-ubuntu1804-11-6-local_11.6.0-510.39.01-1_amd64.deb && \
-    dpkg -i cuda-repo-ubuntu1804-11-6-local_11.6.0-510.39.01-1_amd64.deb && \
+    cuda-ubuntu1804.pin /etc/apt/preferences.d/cuda-repository-pin-600 && \
+    wget https://developer.download.nvidia.com/compute/cuda/11.6.1/local_installers/cuda-repo-ubuntu1804-11-6-local_11.6.1-510.47.03-1_amd64.deb && \
+    dpkg -i cuda-repo-ubuntu1804-11-6-local_11.6.1-510.47.03-1_amd64.deb && \
     apt-key add /var/cuda-repo-ubuntu1804-11-6-local/7fa2af80.pub
+
 RUN apt-get update && apt-get install -y --no-install-recommends --allow-unauthenticated \
     cuda-drivers && \
     apt-get autoclean -y && apt-get autoremove -y  && rm -rf /var/lib/apt/lists/* 
 
-
 # (2) Configurate Xorg
 # (2-1) Install some necessary softwares
 #