diff --git a/Dockerfile b/Dockerfile
index bb3ff91..d2e3836 100644
--- a/Dockerfile
+++ b/Dockerfile
@@ -94,7 +94,7 @@
         libc6 \
         pkg-config \
         libelf-dev \
-        gnupg patch && \
+        gnupg patch xinit && \
     rm -rf /var/lib/apt/lists/*
 
 #        libc6:i386 \
@@ -102,19 +102,22 @@
 # After this installation, command Xorg and xinit can be used in the container
 # if you need full ubuntu desktop environment, the line below should be added.
         # ubuntu-desktop \
-RUN apt-get update && apt-get install -y xinit && \
-    rm -rf /var/lib/apt/lists/*
+#RUN apt-get update && apt-get install -y xinit && \
+#    rm -rf /var/lib/apt/lists/*
 # 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 && \
-   apt-key adv --fetch-keys https://developer.download.nvidia.com/compute/cuda/repos/ubuntu1804/x86_64/7fa2af80.pub && \
-   add-apt-repository "deb https://developer.download.nvidia.com/compute/cuda/repos/ubuntu1804/x86_64/ /" && \
-   apt-get update && apt-get install -y --no-install-recommends \
-   cuda-drivers && \
-   rm -rf /var/lib/apt/lists/*
+    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.4.0/local_installers/cuda-repo-ubuntu1804-11-4-local_11.4.0-470.42.01-1_amd64.deb && \
+    dpkg -i cuda-repo-ubuntu1804-11-4-local_11.4.0-470.42.01-1_amd64.deb && \
+    apt-key add /var/cuda-repo-ubuntu1804-11-4-local/7fa2af80.pub
+   
+RUN apt-get update && \
+    apt-get install -y --no-install-recommends \
+    cuda-drivers && \
+    rm -rf /var/lib/apt/lists/*
 
 
 # (2) Configurate Xorg
@@ -268,4 +271,5 @@
 COPY scripts/ /home/$USERNAME/scripts/
 EXPOSE 5900
 EXPOSE 8081 
+
 CMD ["bash", "/startup.sh"]