Commit ·
24bd825
1
Parent(s): 25f6ae6
Fix torchaudio CUDA mismatch: install via PyTorch cu121 index in pre-install.sh
Browse filestorchaudio 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>
- pre-install.sh +2 -0
- 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
|
|
|