25 Commits

Author SHA1 Message Date
wjunLu
d30cbcdfa5
[ci] Add workflow for building NPU image (#8546)
Co-authored-by: Yaowei Zheng <hiyouga@buaa.edu.cn>
2025-07-04 20:56:59 +08:00
Yaowei Zheng
cecba57b3e
[assets] fix npu docker (#8298) 2025-06-05 19:09:20 +08:00
Yaowei Zheng
81c4d9bee6
[assets] update docker files (#8291) 2025-06-04 23:30:46 +08:00
hoshi-hiyouga
a4048b7bb6
[assets] fix docker images (#8203) 2025-05-28 22:26:05 +08:00
hoshi-hiyouga
55d37dc472
[assets] update Dockerfile (#8201) 2025-05-28 20:20:59 +08:00
hoshi-hiyouga
07f792148e
[assets] fix docker image (#8180) 2025-05-27 19:01:31 +08:00
hoshi-hiyouga
519ac92803
[assets] update docker files (#8176) 2025-05-27 18:15:23 +08:00
hoshi-hiyouga
5a0fd22c05 [assets] update readme (#7209)
Former-commit-id: cdf8fc647819523eca8139ba62a510c3078b694d
2025-03-07 17:27:49 +08:00
leo-pony
e86cb8a4fa [npu] update cann base image and torch 2.4 (#7061)
* Update base npu container image version:The Python version required for Hugging Face Transformers is >= python3.10

* Fix the bug: arg type of INSTALL_DEEPSPEED shoud been string now.

* Update Ascend CANN, CANN-Kernel and corresponding torch and torch-npu version

* Upgrade torch-npu needs packages' version: torch==2.1.0 and torch-npu==2.4.0.post2

Former-commit-id: acc52e0fe75d59c2581ef45346ebfbffe45f762b
2025-02-25 23:32:01 +08:00
hiyouga
a02a140840 update wechat
Former-commit-id: cd1433650653810f7934c65cb1de91052eb73dcf
2025-01-04 07:59:57 +00:00
shibingli@yeah.net
c76c33ddb1 Add ARG HTTP_PROXY in Dockerfile to support HTTP proxy during image building.
Former-commit-id: f1d76786e094562f6f095a0b56c9c6cd32e2fa5e
2024-12-27 18:31:14 +08:00
XYZliang
1c1e48a570 Increase shm_size to 16GB in docker-compose.yml to optimize shared memory allocation for large-scale model fine-tuning tasks.
This pull request increases the shm_size parameter in docker-compose.yml to 16GB. The goal is to enhance the LLaMA-Factory framework’s performance for large model fine-tuning tasks by providing sufficient shared memory for efficient data loading and parallel processing.

This PR also addresses the issues discussed in [this comment](https://github.com/hiyouga/LLaMA-Factory/issues/4316#issuecomment-2466270708) regarding Shared Memory Limit error.


Former-commit-id: 64414905a3728abf3c51968177ffc42cfc653310
2024-11-13 10:13:59 +08:00
hiyouga
e90a1199da tiny fix
Former-commit-id: 3af57795dda5d236200bad4aa3f2e29ae8930fe2
2024-10-11 23:51:54 +08:00
MengqingCao
a620fa8be3 update npu base image
Former-commit-id: b3f4acd1b4636d15afdfde61d77af521fff39b19
2024-08-21 09:12:38 +00:00
hiyouga
4b1ab6c83d update npu docker
Former-commit-id: 93e6fbb37ddb60ed356d0acc720463a8cd1a8baa
2024-06-30 21:05:31 +08:00
hiyouga
7c488cea57 tiny fix
Former-commit-id: e44a4f07f09bbee55c10ccee91dd858256c36054
2024-06-27 20:14:48 +08:00
hoshi-hiyouga
ffe7cda93d Update Dockerfile
Former-commit-id: 5fcd33fd4f08f5e46c1df541f188d13faefa7c8e
2024-06-27 19:57:40 +08:00
hoshi-hiyouga
030dda5861 Update Dockerfile
Former-commit-id: 9624af1df3e4b260b37ad10e2a9be8e69ab066df
2024-06-27 19:51:25 +08:00
fanjunliang
9e93ff9479 support docker-npu-[amd64|arm64] build
Former-commit-id: bdda0827b35cbb6005f10aa661fe6f3940b598d8
2024-06-27 15:25:12 +08:00
fanjunliang
e7bd3ab6c3 fix torch-npu dependency
Former-commit-id: 8096f94a7db3e357c52906c83cad9f3c360bad7e
2024-06-26 18:21:42 +08:00
MengqingCao
7c7d6614d8 fix docker-compose path
Former-commit-id: 106647a99d36c4a624511edf651cf5f7624c4c9a
2024-06-26 02:15:00 +00:00
hiyouga
dafc9268bc fix #4419
Former-commit-id: efb81b25ecd5cb9f4cfda8f2da8b159e4ab26a90
2024-06-25 01:51:29 +08:00
hoshi-hiyouga
af5b2b9299 Update docker-compose.yml
Former-commit-id: 721acd87688b3edd41d2e1890971eb387afc1b6e
2024-06-25 00:54:28 +08:00
hoshi-hiyouga
6cd45e95f7 Update Dockerfile
Former-commit-id: 3af936a76d726c73d14ff6cf8c3088bda5b904c3
2024-06-25 00:50:34 +08:00
MengqingCao
3b499948a5 update docker files
1. add docker-npu (Dockerfile and docker-compose.yml)
  2. move cuda docker to docker-cuda and tiny changes to adapt to the new path


Former-commit-id: d7207e8ad10c7df6dcb1f5e59ff8eb06f9d77e67
2024-06-24 10:57:36 +00:00