大模型微调工具LLaMA-Factory docker安装、训练

时间:2024-06-03 18:58:59

参考:
https://github.com/hiyouga/LLaMA-Factory

报错解决:

1)Docker 构建报错 RuntimeError: can’t start new thread:
https://github.com/hiyouga/LLaMA-Factory/issues/3859
修改后的Dockerfile:

FROM nvcr.io/nvidia/pytorch:24.01-py3
  
WORKDIR /app

COPY requirements.txt /app/

RUN pip config --user set global.progress_bar off

RUN pip install  -r requirements.txt -