diff --git a/Dockerfile b/Dockerfile
index cadf683..90b7a63 100644
--- a/Dockerfile
+++ b/Dockerfile
@@ -9,9 +9,7 @@ RUN pip3 install Cython
 RUN pip3 install --upgrade pip
 
 
-COPY ./torch-whl/torch-1.7.0+cu101-cp38-cp38-linux_x86_64.whl /torch-whl/torch-1.7.0+cu101-cp38-cp38-linux_x86_64.whl
-COPY ./torch-whl/torchvision-0.8.0-cp38-cp38-linux_x86_64.whl /torch-whl/torchvision-0.8.0-cp38-cp38-linux_x86_64.whl
+COPY ./torch-whl/torch-1.6.0+cu101-cp38-cp38-linux_x86_64.whl /torch-whl/torch-1.6.0+cu101-cp38-cp38-linux_x86_64.whl
 
 
-RUN pip3 install ./torch-whl/torch-1.7.0+cu101-cp38-cp38-linux_x86_64.whl
-RUN pip3 install ./torch-whl/torchvision-0.8.0-cp38-cp38-linux_x86_64.whl
+RUN pip3 install ./torch-whl/torch-1.6.0+cu101-cp38-cp38-linux_x86_64.whl