BoxOfColors Claude Sonnet 4.6 commited on
Commit
24bd825
·
1 Parent(s): 25f6ae6

Fix torchaudio CUDA mismatch: install via PyTorch cu121 index in pre-install.sh

Browse files

torchaudio from plain pip installs CPU/wrong-CUDA build causing
libcudart.so.13 error on ZeroGPU. Install via the PyTorch CUDA 12.1
wheel index to match the pre-installed torch on the worker.

Co-Authored-By: Claude Sonnet 4.6 <noreply@anthropic.com>

Files changed (2) hide show
  1. pre-install.sh +2 -0
  2. requirements.txt +0 -1
pre-install.sh CHANGED
@@ -1,2 +1,4 @@
1
  pip install -U setuptools
2
  mim install mmcv>=2.0.0
 
 
 
1
  pip install -U setuptools
2
  mim install mmcv>=2.0.0
3
+ # Install torchaudio matching the pre-installed PyTorch/CUDA build on ZeroGPU
4
+ pip install torchaudio --index-url https://download.pytorch.org/whl/cu121
requirements.txt CHANGED
@@ -16,4 +16,3 @@ spaces==0.48.1
16
  gradio==5.25.0
17
  pydantic==2.8.2
18
  av
19
- torchaudio
 
16
  gradio==5.25.0
17
  pydantic==2.8.2
18
  av