t29mato commited on
Commit
436b98c
·
1 Parent(s): fb7addb

Use mmcv-full with MMCV_WITH_OPS=0 (skip C++ extension build)

Browse files

C++ ops (NMS, RoIAlign) are already patched to torchvision in
chartdete_infer.py, so no native extensions needed.

Files changed (1) hide show
  1. Dockerfile +4 -4
Dockerfile CHANGED
@@ -16,10 +16,10 @@ RUN pip install --no-cache-dir \
16
  torchvision==0.14.1+cpu \
17
  -f https://download.pytorch.org/whl/cpu/torch_stable.html
18
 
19
- # Install mmcv-full pre-built wheel
20
- RUN pip install --no-cache-dir \
21
- mmcv-full==1.7.2 \
22
- -f https://download.openmmlab.com/mmcv/dist/cpu/torch1.13.0/index.html
23
 
24
  # Install remaining dependencies
25
  RUN pip install --no-cache-dir \
 
16
  torchvision==0.14.1+cpu \
17
  -f https://download.pytorch.org/whl/cpu/torch_stable.html
18
 
19
+ # Install mmcv-full from source WITHOUT C++ ops compilation
20
+ # The C++ ops (NMS, RoIAlign) are patched to use torchvision in chartdete_infer.py
21
+ ENV MMCV_WITH_OPS=0
22
+ RUN pip install --no-cache-dir mmcv-full==1.7.2
23
 
24
  # Install remaining dependencies
25
  RUN pip install --no-cache-dir \