-
Notifications
You must be signed in to change notification settings - Fork 615
add ascend_a3 Dockerfile #4097
New issue
Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.
By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.
Already on GitHub? Sign in to your account
base: main
Are you sure you want to change the base?
add ascend_a3 Dockerfile #4097
Conversation
docker/Dockerfile.ascend_a3
Outdated
| FROM ${ASCEND_DEVICE}_base AS builder | ||
| ENV LMDEPLOY_TARGET_DEVICE=ascend | ||
| WORKDIR /opt/lmdeploy | ||
| COPY lmdeploy /opt/lmdeploy |
There was a problem hiding this comment.
Choose a reason for hiding this comment
The reason will be displayed to describe this comment to others. Learn more.
copy . /opt/lmdeploy
There was a problem hiding this comment.
Choose a reason for hiding this comment
The reason will be displayed to describe this comment to others. Learn more.
I will refactor it.
docker/Dockerfile.ascend_a3
Outdated
| ENV DEVICE=ascend | ||
| ENV TORCH_DEVICE_BACKEND_AUTOLOAD=0 | ||
| WORKDIR /opt/dlinfer | ||
| COPY dlinfer /opt/dlinfer |
There was a problem hiding this comment.
Choose a reason for hiding this comment
The reason will be displayed to describe this comment to others. Learn more.
Can we use pip install dlinfer?
There was a problem hiding this comment.
Choose a reason for hiding this comment
The reason will be displayed to describe this comment to others. Learn more.
Of course, I will modify the installation method.
DOCKER_BUILDKIT=1 docker build --build-arg ASCEND_DEVICE=ascend_a3 --build-arg DLINFER_TAG=v0.2.3.post2 --build-arg LMDEPLOY_TAG=v0.10.2 --network=host -t lmdeploy_dlinfer:a3 -f Dockerfile.ascend_a3 .