Commit Graph

15 Commits

Author SHA1 Message Date
shibingli@yeah.net
f1d76786e0 Add ARG HTTP_PROXY in Dockerfile to support HTTP proxy during image building. 2024-12-27 18:31:14 +08:00
XYZliang
64414905a3 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.
2024-11-13 10:13:59 +08:00
hiyouga
3af57795dd tiny fix 2024-10-11 23:51:54 +08:00
MengqingCao
b3f4acd1b4 update npu base image 2024-08-21 09:12:38 +00:00
hiyouga
93e6fbb37d update npu docker 2024-06-30 21:05:31 +08:00
hiyouga
e44a4f07f0 tiny fix 2024-06-27 20:14:48 +08:00
hoshi-hiyouga
5fcd33fd4f Update Dockerfile 2024-06-27 19:57:40 +08:00
hoshi-hiyouga
9624af1df3 Update Dockerfile 2024-06-27 19:51:25 +08:00
fanjunliang
bdda0827b3 support docker-npu-[amd64|arm64] build 2024-06-27 15:25:12 +08:00
fanjunliang
8096f94a7d fix torch-npu dependency 2024-06-26 18:21:42 +08:00
MengqingCao
106647a99d fix docker-compose path 2024-06-26 02:15:00 +00:00
hiyouga
efb81b25ec fix #4419 2024-06-25 01:51:29 +08:00
hoshi-hiyouga
721acd8768 Update docker-compose.yml 2024-06-25 00:54:28 +08:00
hoshi-hiyouga
3af936a76d Update Dockerfile 2024-06-25 00:50:34 +08:00
MengqingCao
d7207e8ad1 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
2024-06-24 10:57:36 +00:00