diff --git a/Dockerfile b/Dockerfile index be2e1612dc2f2d52190b0b7f7e57c78890311e7d..ca057b4e13bb955b44b70ccbde9afd492bf5e667 100644 --- a/Dockerfile +++ b/Dockerfile @@ -44,13 +44,13 @@ USER ${USER} RUN pip install numpy cython --no-cache-dir COPY --chown=1001:1001 requirements.txt requirements.txt RUN pip install --no-cache-dir -r requirements.txt -RUN pip install mmcv-full -f https://download.openmmlab.com/mmcv/dist/cu101/torch1.8.0/index.html +RUN pip install mmcv-full==1.4.0 -f https://download.openmmlab.com/mmcv/dist/cu101/torch1.8.0/index.html #RUN apt-get install jupyter-notebook # (optional) # (either add below as dependency in requirements.txt or uncomment below line) -# RUN pip install -U git+https://github.com/open-mmlab/mmdetection.git@v2.3.0 +# RUN pip install -U git+https://github.com/open-mmlab/mmdetection.git@v2.12.0 # Copy user files COPY --chown=1001:1001 . ${HOME_DIR} diff --git a/requirements.txt b/requirements.txt index 23dd9c5d800c8d91dee461e84dde0a12595109c1..0d02f8f1593c0bdb7f2dd838ebb6d14779f39760 100644 --- a/requirements.txt +++ b/requirements.txt @@ -11,4 +11,4 @@ pyyaml==5.1 timeout_decorator git+https://github.com/AIcrowd/coco.git#subdirectory=PythonAPI -git+https://github.com/open-mmlab/mmdetection.git@v2.3.0 +git+https://github.com/open-mmlab/mmdetection.git@v2.12.0