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

Fix: inline MMCV_WITH_OPS=0 in RUN + --no-build-isolation

Browse files
Files changed (1) hide show
  1. Dockerfile +2 -4
Dockerfile CHANGED
@@ -16,10 +16,8 @@ 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 \
 
16
  torchvision==0.14.1+cpu \
17
  -f https://download.pytorch.org/whl/cpu/torch_stable.html
18
 
19
+ # Install mmcv-full WITHOUT C++ ops (ops are patched to torchvision at runtime)
20
+ RUN MMCV_WITH_OPS=0 FORCE_CUDA=0 pip install --no-cache-dir --no-build-isolation mmcv-full==1.7.2
 
 
21
 
22
  # Install remaining dependencies
23
  RUN pip install --no-cache-dir \