| 123456789101112131415161718192021222324252627 |
- FROM pytorch/pytorch:1.12.1-cuda11.3-cudnn8-runtime
- RUN ln -sf /usr/share/zoneinfo/Asia/Shanghai /etc/localtime && echo 'Asia/Shanghai' > /etc/timezone
- WORKDIR /pose
- RUN sed -i 's/archive.ubuntu.com/mirrors.nju.edu.cn/g' /etc/apt/sources.list && \
- sed -i 's/security.ubuntu.com/mirrors.nju.edu.cn/g' /etc/apt/sources.list && \
- apt update && \
- apt install -y libgl1-mesa-glx libglib2.0-dev libmysqlclient-dev && \
- rm -rf /var/lib/apt/lists/*
- COPY requirements.txt .
- RUN pip3 install --no-cache-dir -i https://pypi.tuna.tsinghua.edu.cn/simple -r requirements.txt
- # RUN python -m pip install detectron2 --no-cache-dir -i https://mirror.nju.edu.cn/pypi/web/simple -f https://dl.fbaipublicfiles.com/detectron2/wheels/cpu/torch1.10/index.html
- VOLUME /data
- COPY . .
- CMD ["gunicorn", "app:app"]
|
