why does install guide use stone age pytorch?
#15
by Andyx1976 - opened
"pip install torch==2.5.1+cu124 torchvision==0.20.1+cu124 torchaudio==2.5.1+cu124" is incredibly ancient. And does not work with any rtx50. Is there a reason i'm missing why it is used there? Does it need it?
Thanks for pointing this out. torch==2.5.1+cu124 is not a hard requirement; it was one of the dependency combinations we had previously validated.
We have tested the following combinations on NVIDIA A100:
- PyTorch 2.8.0 + cu126 + flash-attn 2.8.3
- PyTorch 2.5.1 + cu124 + flash-attn 2.6.3
The README has now been updated with installation instructions for the newer PyTorch 2.8.0 + cu126 setup. Please fetch the latest version of the repository and follow the updated instructions there.