17 lines
579 B
Docker
17 lines
579 B
Docker
FROM python:3.7.13-slim
|
|
|
|
WORKDIR /app
|
|
ADD . /app/
|
|
|
|
RUN pip install --upgrade pip -i https://pypi.tuna.tsinghua.edu.cn/simple --no-cache-dir
|
|
RUN pip install torch torchvision --extra-index-url https://download.pytorch.org/whl/cu113 --no-cache-dir
|
|
RUN pip install -r requirements.txt -i https://pypi.tuna.tsinghua.edu.cn/simple --no-cache-dir
|
|
RUN mim install mmcv-full
|
|
RUN pip install -e ./text2image/CLIP/
|
|
RUN pip install -e ./backbone/mmpose/
|
|
RUN pip install -e ./ocr/tr/
|
|
RUN rm -rf ./text2image/CLIP/
|
|
RUN rm -rf ./ocr/tr/
|
|
RUN rm -rf ./backbone/mmpose/
|
|
|
|
CMD ["python3", "run.py"] |