From b1069b4521caa9082a415b0c64abd2d42e98dd8c Mon Sep 17 00:00:00 2001 From: Jamie Pine Date: Mon, 16 Mar 2026 23:23:55 -0700 Subject: [PATCH] upgrade CUDA backend build from cu121 to cu126 cu121 only ships kernels up to SM 9.0 (Ada Lovelace). RTX 50-series (Blackwell, SM 12.0) and RTX 6000 Pro need cu126 which includes SM 12.0 support while remaining backward compatible with older GPUs. Closes #289 --- .github/workflows/release.yml | 6 +++--- 1 file changed, 3 insertions(+), 3 deletions(-) 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: |