diff --git a/Dockerfile.gpu b/Dockerfile.gpu
index 13d763d..87ce9e8 100644
--- a/Dockerfile.gpu
+++ b/Dockerfile.gpu
@@ -1,4 +1,4 @@
-FROM nvidia/cuda:11.8.0-cudnn8-runtime-ubuntu18.04 AS python-ubuntu
+FROM nvidia/cuda:10.2-cudnn8-runtime-ubuntu18.04 AS python-ubuntu
 
 RUN apt-get update
 RUN apt-get install -y build-essential curl python3 python3-pip python3-venv
@@ -10,7 +10,10 @@ RUN python3 -m pip install wheel
 
 FROM python-ubuntu
 
-RUN apt-get update &&  apt-get install liblapack-dev libpng-dev libfreetype6-dev libqhull-dev git pkg-config portaudio19-dev swig libpulse-ocaml-dev gfortran libopenblas-dev libatlas-base-dev -y
+RUN apt-get update &&  apt-get install -y  \
+    liblapack-dev libpng-dev libfreetype6-dev libqhull-dev git \
+    pkg-config portaudio19-dev swig libpulse-ocaml-dev gfortran \
+    libopenblas-dev libatlas-base-dev
 
 RUN mkdir /training && pip3 install precise-lite && pip3 install tensorflow-gpu==2.3.1