charlie880624/STDC
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

[Enhancement] Add mmcv arg expansion (#748)

Browse Source 
* Add dynamic mmcv install

* Declare ARG after FROM
This commit is contained in:
Miguel Méndez 2021-08-05 14:09:33 +02:00 committed by GitHub
parent 2c1b3ba211
commit 6394679328
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
1 changed files with 3 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

4
docker/serve/Dockerfile
View File

@ -24,7 +24,9 @@ RUN export FORCE_CUDA=1
RUN pip install torchserve torch-model-archiver RUN pip install torchserve torch-model-archiver
# MMLAB # 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 pip install mmsegmentation==${MMSEG}
RUN useradd -m model-server \ RUN useradd -m model-server \