export TMPDIR=~/tmp pip3 install --upgrade pip pip3 install wheel setuptools cmake ninja build psutil pip3 install torch==2.6.0 torchvision==0.21.0 torchaudio==2.6.0 --index-url https://download.pytorch.org/whl/cu124 pip3 install flash-attn --no-build-isolation