Skip to content

[Fix] Update mmcv installation in dockerfile #860

New issue

Have a question about this project? # for a free GitHub account to open an issue and contact its maintainers and the community.

#

By clicking “#”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.

Already on GitHub? # to your account

Merged
merged 4 commits into from
Sep 11, 2021
Merged

[Fix] Update mmcv installation in dockerfile #860

Changes from 1 commit
Commits
Show all changes
4 commits
Select commit Hold shift + click to select a range
40c6d06
update dockerfile
Junjun2016 Sep 8, 2021
f650085
Update docker/Dockerfile
Junjun2016 Sep 9, 2021
0e78a23
add ARG
Junjun2016 Sep 9, 2021
f1056b4
change dash to bash
Junjun2016 Sep 9, 2021
File filter

Filter by extension

Filter by extension
Conversations
Failed to load comments.
Loading
Jump to
Jump to file
Failed to load files.
Loading
Diff view
Diff view
8 changes: 6 additions & 2 deletions docker/Dockerfile
View file
Open in desktop
Original file line number Diff line number Diff line change
@@ -1,6 +1,7 @@
ARG PYTORCH="1.6.0"
ARG CUDA="10.1"
ARG CUDNN="7"
ARG MMCV="1.3.12"

FROM pytorch/pytorch:${PYTORCH}-cuda${CUDA}-cudnn${CUDNN}-devel

Copy link
Contributor

Choose a reason for hiding this comment

The reason will be displayed to describe this comment to others. Learn more.

Hello.When I try to build docker image
/docker$ docker build -t mmseg .
I get a error
Error log:
Step 10/16 : RUN conda clean --all
---> Using cache
---> 025c6545676b
Step 11/16 : RUN ["/bin/bash", "-c", "pip install mmcv-full==${MMCV} -f https://download.openmmlab.com/mmcv/dist/cu${CUDA//./}/torch${PYTORCH}/index.html"]
---> Running in 24b4e9123299
Looking in links: https://download.openmmlab.com/mmcv/dist/cu/torch/index.html
ERROR: Could not find a version that satisfies the requirement mmcv-full== (from versions: 1.0rc0, 1.0rc2, 1.0.0, 1.0.1, 1.0.2, 1.0.3, 1.0.4, 1.0.5, 1.1.0, 1.1.1, 1.1.2, 1.1.3, 1.1.4, 1.1.5, 1.1.6, 1.2.0, 1.2.1, 1.2.2, 1.2.3, 1.2.4, 1.2.5, 1.2.6, 1.2.7, 1.3.0, 1.3.1, 1.3.3, 1.3.4, 1.3.5, 1.3.6, 1.3.7, 1.3.8, 1.3.9, 1.3.10, 1.3.11, 1.3.12)
ERROR: No matching distribution found for mmcv-full==
The command '/bin/bash -c pip install mmcv-full==${MMCV} -f https://download.openmmlab.com/mmcv/dist/cu${CUDA//./}/torch${PYTORCH}/index.html' returned a non-zero code: 1

And we can see "Looking in links: https://download.openmmlab.com/mmcv/dist/cu/torch/index.html",script cannt get args ${MMCV},${CUDA//./},${PYTORCH},beacause args workspace.You can see https://yeasy.gitbook.io/docker_practice/image/dockerfile/arg

We can fix dockerfile to resolve problem:

ARG PYTORCH="1.6.0"
ARG CUDA="10.1"
ARG CUDNN="7"

FROM pytorch/pytorch:${PYTORCH}-cuda${CUDA}-cudnn${CUDNN}-devel

ARG PYTORCH="1.6.0"
ARG CUDA="10.1"
ARG CUDNN="7"
ARG MMCV="1.3.12"

Expand All @@ -12,11 +13,14 @@ RUN apt-get update && apt-get install -y git ninja-build libglib2.0-0 libsm6 lib
&& apt-get clean \
&& rm -rf /var/lib/apt/lists/*

# Install mmsegmentation
RUN conda clean --all

RUN pip install mmcv-full==latest+torch1.6.0+cu101 -f https://download.openmmlab.com/mmcv/dist/index.html
# Install MMCV
RUN ["/bin/bash", "-c", "pip install mmcv-full==${MMCV} -f https://download.openmmlab.com/mmcv/dist/cu${CUDA//./}/torch${PYTORCH}/index.html"]

# Install MMSegmentation
RUN git clone https://github.com/open-mmlab/mmsegmentation.git /mmsegmentation
WORKDIR /mmsegmentation
ENV FORCE_CUDA="1"
RUN pip install -r requirements.txt
RUN pip install --no-cache-dir -e .