安装 X-R1 需分三步完成环境搭建:
- 硬件准备:至少 1 张 NVIDIA GPU(推荐 4 张 3090),16GB 内存,支持 AVX 指令集的 CPU
- 基础依赖安装:
sudo apt update
sudo apt install git python3 python3-pip build-essential
pip install -r requirements.txt
pip install accelerate - 项目配置:
- 克隆仓库:
git clone https://github.com/dhcode-cpp/X-R1.git
- 修改 recipes/zero3.yaml 中的 num_processes(GPU 数量减 1)
- 通过
accelerate config
验证环境
- 克隆仓库:
关键注意事项:
- CUDA 版本必须与 GPU 驱动匹配
- Windows 用户需启用 WSL2
- 首次运行时建议先测试 0.5B 小模型
本答案来源于文章《X-R1:在普通设备中低成本训练0.5B模型》