File tree Expand file tree Collapse file tree
Expand file tree Collapse file tree Original file line number Diff line number Diff line change @@ -36,6 +36,13 @@ dependencies = [
3636[project .optional-dependencies ]
3737whisper = [
3838 " whisper-live>=0.9.0" ,
39+ # CTranslate2 (faster-whisper's backend) loads CUDA via the system
40+ # dynamic loader and can't find torch's bundled copies under
41+ # site-packages/nvidia/. These wheels install the libraries in a
42+ # location CTranslate2 actually searches; without them GPU inference
43+ # dies with "Library libcublas.so.12 is not found".
44+ " nvidia-cublas-cu12; sys_platform == 'linux'" ,
45+ " nvidia-cudnn-cu12; sys_platform == 'linux'" ,
3946]
4047dev = [
4148 " pytest>=8.0" ,
Original file line number Diff line number Diff line change @@ -10,3 +10,13 @@ torchaudio
1010faster-whisper>=1.2
1111websockets
1212scipy
13+
14+ # CUDA runtime libraries needed by CTranslate2 (faster-whisper's backend).
15+ # torch bundles its own copies under site-packages/nvidia/, but CTranslate2
16+ # uses the system dynamic loader and can't find them there. Without these
17+ # wheels you get at transcription time:
18+ # ERROR: Failed to transcribe audio chunk:
19+ # Library libcublas.so.12 is not found or cannot be loaded
20+ # Skip these only if you're forcing CPU inference (compute_type=int8 on CPU).
21+ nvidia-cublas-cu12
22+ nvidia-cudnn-cu12
You can’t perform that action at this time.
0 commit comments