diff --git a/Dockerfile b/Dockerfile
index fe45773..22c1436 100644
--- a/Dockerfile
+++ b/Dockerfile
@@ -21,7 +21,9 @@
         # Replace 1000 with your user/group id
         usermod  --uid 1000 $USERNAME && \
         groupmod --gid 1000 $USERNAME && \
-        chown -Rf $USERNAME.$USERNAME /home/$USERNAME
+        chown -Rf $USERNAME.$USERNAME /home/$USERNAME && \
+        mkdir -p /home/$USERNAME/.cache  && \
+        chmod -R 777 /home/$USERNAME/.cache
 
 # Error constructing proxy for org.gnome.Terminal:/org/gnome/Terminal/Factory0: Failed to execute child process dbus-launch (No such file or directory)
 # fix by setting LANG https://askubuntu.com/questions/608330/problem-with-gnome-terminal-on-gnome-3-12-2
@@ -93,16 +95,16 @@
                        # this option cannot be used on newer driver
                        # --no-glvnd-egl-client \
                        # --no-glvnd-glx-client \
-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.2.2/local_installers/cuda-repo-ubuntu1804-11-2-local_11.2.2-460.32.03-1_amd64.deb && \
-    dpkg -i cuda-repo-ubuntu1804-11-2-local_11.2.2-460.32.03-1_amd64.deb && \
-    apt-key add /var/cuda-repo-ubuntu1804-11-2-local/7fa2af80.pub && \
-    apt-get update && apt-get install -y --no-install-recommends \
-    cuda-drivers && \
-    rm -rf cuda-repo-ubuntu1804-11-2-local_11.2.2-460.32.03-1_amd64.deb && \
-    rm -rf /var/lib/apt/lists/*
+#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.2.2/local_installers/cuda-repo-ubuntu1804-11-2-local_11.2.2-460.32.03-1_amd64.deb && \
+#    dpkg -i cuda-repo-ubuntu1804-11-2-local_11.2.2-460.32.03-1_amd64.deb && \
+#    apt-key add /var/cuda-repo-ubuntu1804-11-2-local/7fa2af80.pub && \
+#    apt-get update && apt-get install -y --no-install-recommends \
+#    cuda-drivers && \
+#    rm -rf cuda-repo-ubuntu1804-11-2-local_11.2.2-460.32.03-1_amd64.deb && \
+#    rm -rf /var/lib/apt/lists/*
 
 
 # (2) Configurate Xorg