mirror of
https://github.com/open-mmlab/mmpretrain.git
synced 2025-06-03 14:59:18 +08:00
4dd8a86145
* Bump version to v1.0.0rc8 * Apply suggestions from code review Co-authored-by: Yixiao Fang <36138628+fangyixiao18@users.noreply.github.com> * Update README.md --------- Co-authored-by: Yixiao Fang <36138628+fangyixiao18@users.noreply.github.com>
38 lines
776 B
Docker
38 lines
776 B
Docker
ARG PYTORCH="1.12.1"
|
|
ARG CUDA="11.3"
|
|
ARG CUDNN="8"
|
|
FROM pytorch/torchserve:latest-gpu
|
|
|
|
ARG MMPRE="1.0.0rc8"
|
|
|
|
ENV PYTHONUNBUFFERED TRUE
|
|
|
|
ENV HOME="/home/model-server"
|
|
ENV PATH="/opt/conda/bin:$HOME/.local/bin:$PATH"
|
|
RUN export FORCE_CUDA=1
|
|
|
|
# TORCHSEVER
|
|
RUN pip install torchserve torch-model-archiver
|
|
RUN pip install nvgpu
|
|
|
|
# OPEN-MMLAB
|
|
ARG PYTORCH
|
|
ARG CUDA
|
|
RUN pip install openmim
|
|
RUN mim install mmpretrain==${MMPRE}
|
|
RUN mkdir -p $HOME/tmp
|
|
|
|
COPY --chown=model-server entrypoint.sh $HOME/.local/bin/entrypoint.sh
|
|
|
|
RUN chmod +x $HOME/.local/bin/entrypoint.sh
|
|
|
|
COPY --chown=model-server config.properties $HOME/config.properties
|
|
|
|
EXPOSE 8080 8081 8082
|
|
|
|
USER model-server
|
|
WORKDIR $HOME
|
|
ENV TEMP=$HOME/tmp
|
|
ENTRYPOINT ["/home/model-server/.local/bin/entrypoint.sh"]
|
|
CMD ["serve"]
|