Update Dockerfile

Dockerfile

@@ -6,8 +6,16 @@ ENV MKL_THREADING_LAYER=GNU
 # Install dependencies, including OpenBLAS
 RUN apt-get update && apt-get install -y \
     git wget libgl1-mesa-glx libglib2.0-0 ffmpeg libx264-dev \
-    build-essential cmake g++ python3-dev \
-    cuda-toolkit-11-7 libopenblas-dev liblapack-dev
+    build-essential cmake g++ python3-dev
+
+RUN apt-get update && apt-get install -y \
+    software-properties-common && \
+    add-apt-repository -y ppa:graphics-drivers/ppa && \
+    apt-get update && apt-get install -y \
+    cuda-command-line-tools-11-7 \
+    libcusolver-dev-11-7 \
+    libcublas-dev-11-7 \
+    libcudnn8
 
 # Create a user
 RUN useradd -m -u 1000 user