LLaMA-Factory安装教程(具体版)

[复制链接]
发表于 2025-6-3 21:44:24 | 显示全部楼层 |阅读模式

马上注册,结交更多好友,享用更多功能,让你轻松玩转社区。

您需要 登录 才可以下载或查看,没有账号?立即注册

×

本机显卡双3090
使用wsl中ubuntu
torch==2.6.0
conda==24.5.0
cuda==12.4
python==3.12.4(python安装不做赘述,有必要我会另开一篇文章)
一、准备工作

起首,在 https://developer.nvidia.com/cuda-gpus 查看您的 GPU 是否支持CUDA。
包管当前 Linux 版本支持CUDA. 在命令行中输入 
  1. uname -m && cat /etc/*release
复制代码
输出如下,不一定完全一样,雷同即可

查抄是否安装了 gcc . 在命令行中输入 gcc --version ,应当看到雷同的输出

二、pytorch和cuda安装

使用官网版本2.6.0   官网链接:Previous PyTorch Versions   箭头一栏可以看到历史版本


根据其指令安装对应版本
  1. # ROCM 6.1 (Linux only)
  2. pip install torch==2.6.0 torchvision==0.21.0 torchaudio==2.6.0 --index-url https://download.pytorch.org/whl/rocm6.1
  3. # ROCM 6.2.4 (Linux only)
  4. pip install torch==2.6.0 torchvision==0.21.0 torchaudio==2.6.0 --index-url https://download.pytorch.org/whl/rocm6.2.4
  5. # CUDA 11.8
  6. pip install torch==2.6.0 torchvision==0.21.0 torchaudio==2.6.0 --index-url https://download.pytorch.org/whl/cu118
  7. # CUDA 12.4
  8. pip install torch==2.6.0 torchvision==0.21.0 torchaudio==2.6.0 --index-url https://download.pytorch.org/whl/cu124
  9. # CUDA 12.6
  10. pip install torch==2.6.0 torchvision==0.21.0 torchaudio==2.6.0 --index-url https://download.pytorch.org/whl/cu126
  11. # CPU only
  12. pip install torch==2.6.0 torchvision==0.21.0 torchaudio==2.6.0 --index-url https://download.pytorch.org/whl/cpu
复制代码
三、校检torch和cuda版本

安装完查看自己的torch版本
  1. conda list
复制代码

cuda版本则必要进入python
  1. import torch
  2. print(torch.version.cuda)
复制代码

四、LLaMA-Factory 安装

运行以下指令以安装 LLaMA-Factory 及其依赖:
  1. git clone --depth 1 https://github.com/hiyouga/LLaMA-Factory.git
  2. cd LLaMA-Factory
  3. pip install -e ".[torch,metrics]"
复制代码
五、LLaMA-Factory 校验

完成安装后,可以通过使用以下命令。
  1. llamafactory-cli version
复制代码
如果您能成功看到雷同下面的界面,就说明安装成功了。

六、LLaMA-Factory 网页版

LLaMA-Factory 支持通过 WebUI 零代码微调大语言模型。 在完成 安装 后,通过以下指令进入 WebUI:
  1. llamafactory-cli webui
复制代码

其中 http://127.0.0.1:7860就是你的可视化网址。
注:该网址只限于本机使用,如需联网多人服务器使用则需使用内网穿透。
ps.必要更具体的LLaMA-Factory参数详解,请点击关注催更留言,以为内容不错留个赞再走吧,您的承认就是我更新的动力。
回复

使用道具 举报

登录后关闭弹窗

登录参与点评抽奖  加入IT实名职场社区
去登录
快速回复 返回顶部 返回列表