|
@@ -76,7 +76,7 @@ RUN pip install --no-cache-dir --upgrade pip && \ |
|
|
ENV SHELL=/bin/bash |
|
|
ENV SHELL=/bin/bash |
|
|
|
|
|
|
|
|
# install special package |
|
|
# install special package |
|
|
RUN pip install --no-cache-dir mmcls>=0.21.0 mmdet>=2.25.0 decord>=0.6.0 datasets==2.1.0 numpy==1.18.5 ipykernel fairseq |
|
|
|
|
|
|
|
|
RUN pip install --no-cache-dir mmcls>=0.21.0 mmdet>=2.25.0 decord>=0.6.0 datasets==2.1.0 numpy==1.18.5 ipykernel fairseq fasttext https://modelscope.oss-cn-beijing.aliyuncs.com/releases/dependencies/xtcocotools-1.12-cp37-cp37m-linux_x86_64.whl |
|
|
|
|
|
|
|
|
RUN if [ "$USE_GPU" = "True" ] ; then \ |
|
|
RUN if [ "$USE_GPU" = "True" ] ; then \ |
|
|
pip install --no-cache-dir dgl-cu113 dglgo -f https://data.dgl.ai/wheels/repo.html; \ |
|
|
pip install --no-cache-dir dgl-cu113 dglgo -f https://data.dgl.ai/wheels/repo.html; \ |
|
|