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:
```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:
```shell

View File

@ -28,13 +28,13 @@ version = "0.3.11"
"""
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.
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.