From bf8a0afccfc505c4a2d5a820361a7069bb37fc6b Mon Sep 17 00:00:00 2001
From: nbei <nbeiray@gmail.com>
Date: Sun, 27 Sep 2020 16:37:23 +0800
Subject: [PATCH 1/2] update dockerfile with pt1.6

---
 README.md         | 2 +-
 docker/Dockerfile | 3 ++-
 2 files changed, 3 insertions(+), 2 deletions(-)

diff --git a/README.md b/README.md
index 05e8a78822..62e5b77102 100644
--- a/README.md
+++ b/README.md
@@ -11,7 +11,7 @@
 
 MMEditing is an open source image and video editing toolbox based on PyTorch. It is a part of the [OpenMMLab](https://open-mmlab.github.io/) project.
 
-The master branch works with **PyTorch 1.3 to 1.5**.
+The master branch works with **PyTorch 1.3 to 1.6**.
 
 <div align="left">
   <img src="resources/mmediting-demo.jpg"/>
diff --git a/docker/Dockerfile b/docker/Dockerfile
index 93dc25183d..5f2d37a43c 100644
--- a/docker/Dockerfile
+++ b/docker/Dockerfile
@@ -1,4 +1,4 @@
-ARG PYTORCH="1.5"
+ARG PYTORCH="1.6"
 ARG CUDA="10.1"
 ARG CUDNN="7"
 
@@ -17,5 +17,6 @@ RUN conda clean --all
 RUN git clone https://github.com/open-mmlab/mmediting.git /mmediting
 WORKDIR /mmediting
 ENV FORCE_CUDA="1"
+RUN pip install mmcv-full==latest+torch1.6.0+cu101 -f https://download.openmmlab.com/mmcv/dist/index.html
 RUN pip install cython --no-cache-dir
 RUN pip install --no-cache-dir -e .

From cbe829b1ea78311c92a33ed88ee09e08e834e0d3 Mon Sep 17 00:00:00 2001
From: nbei <631557085@qq.com>
Date: Sun, 27 Sep 2020 21:25:38 +0800
Subject: [PATCH 2/2] fix bug for dock-hub

---
 docker/Dockerfile | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

diff --git a/docker/Dockerfile b/docker/Dockerfile
index 5f2d37a43c..86aafc9b94 100644
--- a/docker/Dockerfile
+++ b/docker/Dockerfile
@@ -1,4 +1,4 @@
-ARG PYTORCH="1.6"
+ARG PYTORCH="1.6.0"
 ARG CUDA="10.1"
 ARG CUDNN="7"