Dockerfile

FROM image.sourcefind.cn:5000/dcu/admin/base/pytorch:1.13.1-centos7.6-dtk-23.04-py38-latest

COPY . /tmp/

WORKDIR /tmp

# RUN mkdir -p /root/.cache/torch/hub/checkpoints/

RUN pip install -r requirements.txt -i https://mirrors.aliyun.com/pypi/simple

# RUN cp -r ./extra_models/* /root/.cache/torch/hub/checkpoints/