mamba create -n astra python=3.12
mamba activate astra

pip install torch==2.7.1 torchvision==0.22.1 torchaudio==2.7.1 --index-url https://download.pytorch.org/whl/cu126

git clone https://github.com/zhaisilong/astra
cd astra

pip install -e ".[dev]"

python 1.init_jobs.py
python 2.gen_af3_config.py

afrun predict --data_dir=afdbcp --job_id=1b5n
afrun eval --data_dir=afdbcp --job_id=1b5n