Update to cu124 for torch 2.6.0 support

This commit is contained in:
doctorpangloss 2025-01-31 09:13:47 -08:00
parent 95a12f42e2
commit c1b173e62a
2 changed files with 3 additions and 3 deletions

View File

@ -156,7 +156,7 @@ When using Windows, open the **Windows Powershell** app. Then observe you are at
To enable `torchaudio` support on Windows, install it directly: To enable `torchaudio` support on Windows, install it directly:
```powershell ```powershell
uv pip install torchaudio==2.5.1+cu121 --index-url https://download.pytorch.org/whl/cu121 uv pip install torchaudio==2.6.0+cu124 --index-url https://download.pytorch.org/whl/cu124
``` ```
7. To run the web server: 7. To run the web server:
```shell ```shell

View File

@ -28,13 +28,13 @@ version = "0.3.11"
""" """
The package index to the torch built with AMD ROCm. The package index to the torch built with AMD ROCm.
""" """
amd_torch_index = ("https://download.pytorch.org/whl/rocm6.0", "https://download.pytorch.org/whl/nightly/rocm6.1") amd_torch_index = ("https://download.pytorch.org/whl/rocm6.2", "https://download.pytorch.org/whl/nightly/rocm6.2.4")
""" """
The package index to torch built with CUDA. The package index to torch built with CUDA.
Observe the CUDA version is in this URL. Observe the CUDA version is in this URL.
""" """
nvidia_torch_index = ("https://download.pytorch.org/whl/cu121", "https://download.pytorch.org/whl/nightly/cu124") nvidia_torch_index = ("https://download.pytorch.org/whl/cu124", "https://download.pytorch.org/whl/nightly/cu126")
""" """
The package index to torch built against CPU features. The package index to torch built against CPU features.