安装 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モデルを低コストでトレーニングについて