Skip to content

Instantly share code, notes, and snippets.

@acidsound

acidsound/set_torch.bat

Created March 16, 2026 15:48
Show Gist options

  • Select an option

    Learn more about clone URLs
  • Save acidsound/67125a45230c655850c034878d2cf6d8 to your computer and use it in GitHub Desktop.

Select an option

Learn more about clone URLs
Save acidsound/67125a45230c655850c034878d2cf6d8 to your computer and use it in GitHub Desktop.
Download ZIP
Raw
set_torch.bat
@REM torch 2.9.1 cu130 cp312
uv pip install https://download.pytorch.org/whl/cu130/torch-2.9.1%2Bcu130-cp312-cp312-win_amd64.whl#sha256=cd3232a562ad2a2699d48130255e1b24c07dfe694a40dcd24fad683c752de121
@REM torchvision 0.25.0
uv pip install https://download-r2.pytorch.org/whl/cu130/torchvision-0.24.1%2Bcu130-cp312-cp312-win_amd64.whl#sha256=d31ceaded0d9b737471fa680ccd9e1acb6d5f0f70f03ef3a8d786a99c79da7cf
@REM flash attn 3
uv pip install https://download.pytorch.org/whl/cu129/flash_attn_3-3.0.0-cp39-abi3-win_amd64.whl
@acidsound
Copy link
Copy Markdown
Author

acidsound commented Mar 16, 2026

uv pip install https://huggingface.co/ussoewwin/Flash-Attention-2_for_Windows/resolve/main/flash_attn-2.8.3%2Bcu130torch2.9.1cxx11abiTRUE-cp312-cp312-win_amd64.whl

Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment