fix an error that mmcv version does not match when when import mmpose (…

…open-mmlab#1234) Co-authored-by: zhangwenjun <[email protected]>
shuheilocale · Mar 11, 2022 · 735bba7 · 735bba7
1 parent 16a0522
commit 735bba7
Showing 1 changed file with 2 additions and 1 deletion.
diff --git a/docker/Dockerfile b/docker/Dockerfile
@@ -17,7 +17,8 @@ RUN pip install cython
 RUN pip install xtcocotools
 
 # Install MMCV
-RUN pip install mmcv-full==latest+torch1.6.0+cu101 -f https://download.openmmlab.com/mmcv/dist/index.html
+RUN pip install --no-cache-dir --upgrade pip wheel setuptools
+RUN pip install --no-cache-dir mmcv-full==1.3.17 -f https://download.openmmlab.com/mmcv/dist/cu101/torch1.6.0/index.html
 
 # Install MMPose
 RUN conda clean --all