diff --git a/Dockerfile b/Dockerfile
index d2e3836..13d9955 100644
--- a/Dockerfile
+++ b/Dockerfile
@@ -48,6 +48,8 @@
     apt-get clean && \
     rm -rf /var/lib/apt/lists/*
 
+RUN pip install --upgrade pip setuptools
+
 RUN packages='libsdl2-2.0 xserver-xorg libvulkan1' \
 	&& apt-get update && DEBIAN_FRONTEND=noninteractive apt-get install -y $packages --no-install-recommends \
     && VULKAN_API_VERSION=`dpkg -s libvulkan1 | grep -oP 'Version: [0-9|\.]+' | grep -oP '[0-9|\.]+'` && \
@@ -110,10 +112,10 @@
 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.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
-   
+    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 apt-get update && \
     apt-get install -y --no-install-recommends \
     cuda-drivers && \