diff --git a/docker/Dockerfile b/docker/Dockerfile
index de3c4d1..a6c967f 100644
--- a/docker/Dockerfile
+++ b/docker/Dockerfile
@@ -1,10 +1,10 @@
-FROM nvidia/cudagl:10.0-devel-ubuntu18.04
+FROM nvidia/cudagl:9.2-devel-ubuntu18.04
 ARG cpu_image=0
 ENV CUDNN_VERSION 7.4.1.5
 SHELL ["/usr/bin/env", "bash", "-euxvc"]
 RUN apt-get update && apt-get install -y --no-install-recommends\
-            libcudnn7=$CUDNN_VERSION-1+cuda10.0 \
-            libcudnn7-dev=$CUDNN_VERSION-1+cuda10.0 && \
+            libcudnn7=$CUDNN_VERSION-1+cuda9.2 \
+            libcudnn7-dev=$CUDNN_VERSION-1+cuda9.2 && \
         apt-mark hold libcudnn7
 RUN apt-get update && apt-get install -y --no-install-recommends \
         wget git unzip python3-dev cmake ninja-build libz-dev libssl-dev libboost-dev libglew-dev libtiff-dev python3-venv libjpeg-turbo8-dev \
@@ -21,7 +21,7 @@ RUN python3 -m venv /pix2face/venv;\
     pip install --upgrade pip; \
     pip install -r /pix2face/requirements.txt; \
     if [ $CPU_ONLY -eq 0 ]; then \
-      pip install https://download.pytorch.org/whl/cu100/torch-1.0.0-cp36-cp36m-linux_x86_64.whl; \
+      pip install torch; \
     else \
       pip install https://download.pytorch.org/whl/cpu/torch-1.0.0-cp36-cp36m-linux_x86_64.whl; \
     fi; \