diff --git a/.github/workflows/release.yml b/.github/workflows/release.yml index 8165de0f..cdbec47a 100644 --- a/.github/workflows/release.yml +++ b/.github/workflows/release.yml @@ -189,10 +189,10 @@ jobs: pip install -r backend/requirements.txt pip install --no-deps chatterbox-tts - - name: Install PyTorch with CUDA 12.1 + - name: Install PyTorch with CUDA 12.6 run: | - pip install torch --index-url https://download.pytorch.org/whl/cu121 --force-reinstall --no-deps - pip install torchaudio --index-url https://download.pytorch.org/whl/cu121 + pip install torch --index-url https://download.pytorch.org/whl/cu126 --force-reinstall --no-deps + pip install torchaudio --index-url https://download.pytorch.org/whl/cu126 - name: Verify CUDA support in torch run: |