[Enhancement] Add mmcv arg expansion (#748)
* Add dynamic mmcv install * Declare ARG after FROM
This commit is contained in:
parent
2c1b3ba211
commit
6394679328
@ -24,7 +24,9 @@ RUN export FORCE_CUDA=1
|
||||
RUN pip install torchserve torch-model-archiver
|
||||
|
||||
# MMLAB
|
||||
RUN pip install mmcv-full==${MMCV} -f https://download.openmmlab.com/mmcv/dist/cu101/torch1.6.0/index.html
|
||||
ARG PYTORCH
|
||||
ARG CUDA
|
||||
RUN ["/bin/bash", "-c", "pip install mmcv-full==${MMCV} -f https://download.openmmlab.com/mmcv/dist/cu${CUDA//./}/torch${PYTORCH}/index.html"]
|
||||
RUN pip install mmsegmentation==${MMSEG}
|
||||
|
||||
RUN useradd -m model-server \
|
||||
|
||||
Loading…
x
Reference in New Issue
Block a user