423A35C7/LLaMA-Factory
1
0
Fork 0
mirror of https://github.com/hiyouga/LLaMA-Factory.git synced 2026-08-17 13:35:44 +08:00
Code Issues Packages Projects Releases Wiki Activity

[docker] upgrade NPU images to CANN 9.1 and PyTorch 2.10 (#10729)

Browse Source
This commit is contained in:
xvxuopop
2026-08-10 11:20:41 +08:00
committed by GitHub
parent 63a89710c7
commit 0bbe481e6e
10 changed files with 297 additions and 384 deletions
Download Patch File Download Diff File Expand all files Collapse all files

21
docker/docker-npu/docker-compose.yml
View File

@@ -26,25 +26,26 @@ x-npu-common: &npu-common
services:
llamafactory-a2-ubuntu:
<<: *npu-common
profiles: ["a2-ubuntu"]
build:
<<: *build
args:
<<: *build-args
BASE_IMAGE: quay.io/ascend/cann:9.0.0-910b-ubuntu22.04-py3.11
container_name: llamafactory-a2-ubuntu
image: llamafactory:npu-a2-ubuntu
BASE_IMAGE: quay.io/ascend/cann:9.1.0-910b-ubuntu22.04-py3.12
container_name: llamafactory-910b-ubuntu
image: llamafactory:npu-910b-ubuntu
ports:
- "7860:7860"
- "8000:8000"
llamafactory-a3-ubuntu:
<<: *npu-common
profiles: ["a3"]
profiles: ["a3-ubuntu"]
build:
<<: *build
args:
<<: *build-args
BASE_IMAGE: quay.io/ascend/cann:9.0.0-a3-ubuntu22.04-py3.11
BASE_IMAGE: quay.io/ascend/cann:9.1.0-a3-ubuntu22.04-py3.12
container_name: llamafactory-a3-ubuntu
image: llamafactory:npu-a3-ubuntu
ports:
@@ -53,14 +54,14 @@ services:
llamafactory-a2-openeuler:
<<: *npu-common
profiles: ["openeuler"]
profiles: ["a2-openeuler"]
build:
<<: *build
args:
<<: *build-args
BASE_IMAGE: quay.io/ascend/cann:9.0.0-910b-openeuler24.03-py3.11
container_name: llamafactory-a2-openeuler
image: llamafactory:npu-a2-openeuler
BASE_IMAGE: quay.io/ascend/cann:9.1.0-910b-openeuler24.03-py3.12
container_name: llamafactory-910b-openeuler
image: llamafactory:npu-910b-openeuler
ports:
- "7862:7860"
- "8002:8000"
@@ -72,7 +73,7 @@ services:
<<: *build
args:
<<: *build-args
BASE_IMAGE: quay.io/ascend/cann:9.0.0-a3-openeuler24.03-py3.11
BASE_IMAGE: quay.io/ascend/cann:9.1.0-a3-openeuler24.03-py3.12
container_name: llamafactory-a3-openeuler
image: llamafactory:npu-a3-openeuler
ports: