diff --git a/Dockerfile b/Dockerfile
index c575850..2e612ab 100644
--- a/Dockerfile
+++ b/Dockerfile
@@ -5,7 +5,7 @@ ENV PATH ${ANACONDA}/bin:${CUDA_PATH}/bin:$PATH
 ENV LD_LIBRARY_PATH ${ANACONDA}/lib:${CUDA_PATH}/bin64:$LD_LIBRARY_PATH
 ENV C_INCLUDE_PATH ${CUDA_PATH}/include
 RUN apt-get update && apt-get install -y wget build-essential axel imagemagick
-RUN wget https://repo.continuum.io/archive/Anaconda2-5.0.1-Linux-x86_64.sh -P /tmp
+RUN wget https://repo.continuum.io/archive/Anaconda3-5.0.1-Linux-x86_64.sh -P /tmp
 RUN bash /tmp/Anaconda3-5.0.1-Linux-x86_64.sh -b -p $ANACONDA
 RUN rm /tmp/Anaconda3-5.0.1-Linux-x86_64.sh -rf
 RUN conda install -y pytorch=0.3.0 torchvision cuda90 -c pytorch