Installation¶

Requirements¶

Python: >= 3.12
CUDA: >= 12.1

Install¶

Install from script (quickest)¶

We provide a script to install RAGEN on Linux. This script will install all the dependencies and set up the Conda environment for you.

# Clone the repository
git clone https://github.com/ZihanWang314/RAGEN.git

# Change directory to RAGEN
cd RAGEN

# Setup and download the dataset
bash scripts/setup_ragen.sh

Install from source¶

We also provide manual installation command lines. If setup_ragen.sh fails, you can follow these manual steps in scripts/setup_ragen.md. We recommend you to use Conda for environment management:

Manual Scripts to Setup Environment¶

conda create -n ragen python=3.12 -y
conda activate ragen


git clone git@github.com:ZihanWang314/ragen.git
cd ragen

pip install -e .
pip install torch==2.6.0 --index-url https://download.pytorch.org/whl/cu124

# Optional: to install flash-attn, you may need to install cuda-toolkit first if you don't have
conda install -c "nvidia/label/cuda-12.4.0" cuda-toolkit -y
export CUDA_HOME=$CONDA_PREFIX # /opt/conda/envs/zero
pip3 install flash-attn --no-build-isolation

pip install -r requirements.txt

git submodule init
git submodule update
cd verl
pip install -e .
cd ..