Skip to content

Commit 710212f

Browse files
committed
add missing ctranslate2 libcublas dep
1 parent 59a4268 commit 710212f

2 files changed

Lines changed: 17 additions & 0 deletions

File tree

pyproject.toml

Lines changed: 7 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -36,6 +36,13 @@ dependencies = [
3636
[project.optional-dependencies]
3737
whisper = [
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
]
4047
dev = [
4148
"pytest>=8.0",

requirements/whisper.txt

Lines changed: 10 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -10,3 +10,13 @@ torchaudio
1010
faster-whisper>=1.2
1111
websockets
1212
scipy
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

0 commit comments

Comments
 (0)