mirror of
https://github.com/modelscope/modelscope.git
synced 2026-08-29 10:08:40 +02:00
Backport/release 1.39 to master (#1767)
This commit is contained in:
26
docker/Dockerfile.amd
Normal file
26
docker/Dockerfile.amd
Normal file
@@ -0,0 +1,26 @@
|
||||
FROM {base_image}
|
||||
|
||||
ARG BASE_IMAGE_TAG={base_image_tag}
|
||||
LABEL modelscope.base_image="vllm/vllm-openai-rocm:${BASE_IMAGE_TAG}"
|
||||
|
||||
# Build-time only (ARG does not persist into the image). Aliyun mirror can lag PyPI (~1h).
|
||||
ARG PIP_EXTRA_INDEX_URL=https://pypi.org/simple
|
||||
|
||||
COPY docker/scripts/modelscope_env_init.sh /usr/local/bin/ms_env_init.sh
|
||||
|
||||
ARG CUR_TIME={cur_time}
|
||||
RUN echo "CUR_TIME=${CUR_TIME}" && echo "BASE_IMAGE_TAG=${BASE_IMAGE_TAG}"
|
||||
|
||||
ARG PIP_EXTRA_INDEX_URL=https://pypi.org/simple
|
||||
RUN export PIP_EXTRA_INDEX_URL="${PIP_EXTRA_INDEX_URL}" && \
|
||||
pip config set global.index-url https://mirrors.aliyun.com/pypi/simple && \
|
||||
pip config set install.trusted-host mirrors.aliyun.com && \
|
||||
cd /tmp && GIT_LFS_SKIP_SMUDGE=1 git clone -b {modelscope_branch} --single-branch https://github.com/modelscope/modelscope.git && \
|
||||
cd modelscope && pip install --no-cache-dir . -f https://modelscope.oss-cn-beijing.aliyuncs.com/releases/repo.html && \
|
||||
cd / && rm -fr /tmp/modelscope && pip cache purge
|
||||
|
||||
ENV VLLM_USE_MODELSCOPE=True
|
||||
ENV LMDEPLOY_USE_MODELSCOPE=True
|
||||
ENV MODELSCOPE_CACHE=/mnt/workspace/.cache/modelscope/hub
|
||||
|
||||
SHELL ["/bin/bash", "-c"]
|
||||
Reference in New Issue
Block a user