Skip to content
Merged
Show file tree
Hide file tree
Changes from all commits
Commits
File filter

Filter by extension

Filter by extension


Conversations
Failed to load comments.
Loading
Jump to
Jump to file
Failed to load files.
Loading
Diff view
Diff view
8 changes: 4 additions & 4 deletions .github/workflows/test_docker.yml
Original file line number Diff line number Diff line change
Expand Up @@ -94,10 +94,10 @@ jobs:
- name: Build Docker image
run: |
docker build . -t lmdeploy:ascend -f docker/Dockerfile.ascend --build-arg PYTHON_VERSION=${PYTHON_VERSION}
- name: Test image with lmdeploy check_env
run: |
docker images
docker run --rm lmdeploy:ascend lmdeploy check_env
# - name: Test image with lmdeploy check_env
# run: |
# docker images
# docker run --rm lmdeploy:ascend lmdeploy check_env
- name: Dive
uses: MaxymVlasov/dive-action@v1.5.0
with:
Expand Down
8 changes: 3 additions & 5 deletions docker/Dockerfile.ascend
Original file line number Diff line number Diff line change
@@ -1,6 +1,6 @@
ARG CANN_TAG=8.1.RC1-python3.11-oe2403lts
ARG CANN_TAG=8.2.rc1-a3-ubuntu22.04-py3.11

FROM openeuler/cann:${CANN_TAG} AS builder
FROM ascendai/cann:${CANN_TAG} AS builder
ARG PYTHON_VERSION=3.10

RUN --mount=type=cache,target=/root/.cache \
Expand All @@ -17,13 +17,11 @@ RUN --mount=type=cache,target=/root/.cache \
pip install --upgrade pip build && \
python -m build -w -o /wheels -v .

FROM openeuler/cann:${CANN_TAG} AS final
FROM ascendai/cann:${CANN_TAG} AS final
COPY --from=builder /root/.local /root/.local
COPY --from=builder /opt/py3 /opt/py3
ENV PATH=/opt/py3/bin:$PATH

RUN --mount=type=cache,target=/root/.cache \
--mount=type=cache,target=/wheels,from=builder,source=/wheels \
pip install /wheels/*.whl

ENTRYPOINT []
Loading