diff --git a/Dockerfile b/Dockerfile
index 5d365791..7e21e827 100644
--- a/Dockerfile
+++ b/Dockerfile
@@ -11,13 +11,14 @@ RUN apt-get update \
 # Install python miniconda3 + requirements
 ENV MINICONDA_HOME="/opt/miniconda"
 ENV PATH="${MINICONDA_HOME}/bin:${PATH}"
-RUN curl -o Miniconda3-latest-Linux-x86_64.sh https://repo.continuum.io/miniconda/Miniconda3-latest-Linux-x86_64.sh \
+RUN curl -L -o Miniconda3-latest-Linux-x86_64.sh https://repo.continuum.io/miniconda/Miniconda3-latest-Linux-x86_64.sh \
     && chmod +x Miniconda3-latest-Linux-x86_64.sh \
     && ./Miniconda3-latest-Linux-x86_64.sh -b -p "${MINICONDA_HOME}" \
     && rm Miniconda3-latest-Linux-x86_64.sh
 COPY environment.yml environment.yml
+RUN conda install python=3.8.5
 RUN conda env update -n=root --file=environment.yml
-RUN conda clean -y -i -l -p -t && \
+RUN conda clean -y -i -p -t && \
     rm environment.yml
 
 # Clone deep image prior repository
diff --git a/environment.yml b/environment.yml
index 6c85cb42..cf4375e8 100644
--- a/environment.yml
+++ b/environment.yml
@@ -1,17 +1,17 @@
 name: deep-image-prior
 channels:
-- pytorch
-- defaults
+  - pytorch
+  - defaults
 dependencies:
-- jupyter
-- nb_conda
-- numpy
-- pyyaml
-- mkl
-- setuptools
-- cmake
-- cffi
-- pytorch=0.4
-- matplotlib
-- scikit-image
-- torchvision
+  - jupyter
+  - nb_conda
+  - numpy
+  - pyyaml
+  - mkl
+  - setuptools
+  - cmake
+  - cffi
+  - pytorch
+  - matplotlib
+  - scikit-image
+  - torchvision
\ No newline at end of file