diff --git a/Dockerfile b/Dockerfile
index 3793e62..3a7163f 100644
--- a/Dockerfile
+++ b/Dockerfile
@@ -1,4 +1,4 @@
-FROM nvidia/cuda:11.7.0-cudnn8-devel-ubuntu18.04
+FROM nvidia/cuda:10.1-cudnn7-devel-ubuntu18.04
 
 # encoding
 RUN echo "LC_ALL=en_US.UTF-8" >> /etc/environment
@@ -9,7 +9,8 @@ RUN echo "LANG=en_US.UTF-8" > /etc/locale.conf
 # fix (tzdata)
 ARG DEBIAN_FRONTEND=noninteractive
 
-
+RUN apt-key adv --fetch-keys https://developer.download.nvidia.com/compute/cuda/repos/ubuntu1804/x86_64/3bf863cc.pub
+RUN apt-key adv --fetch-keys https://developer.download.nvidia.com/compute/machine-learning/repos/ubuntu1804/x86_64/7fa2af80.pub
 RUN apt-get -y update && \
     apt-get -y upgrade && \
     apt-get install -y python3.8 python3-pip