Add cudnn and cublas dependencies for GPU support

These are faster-whisper dependencies, see https://github.com/SYSTRAN/faster-whisper/blob/d57c5b40b06e59ec44240d93485a95799548af50/README.md
beeldengeluid · Sep 26, 2024 · 8bf366d · 8bf366d
1 parent 2f1ea0c
commit 8bf366d
Show file tree

Hide file tree

Showing 2 changed files with 42 additions and 1 deletion.
diff --git a/poetry.lock b/poetry.lock
diff --git a/pyproject.toml b/pyproject.toml
@@ -10,6 +10,8 @@ readme = "README.md"
 python = "^3.10"
 dane = "^0.4.3"
 faster-whisper = "^1.0.1"
+nvidia-cudnn-cu12 = "^8"
+nvidia-cublas-cu12 = "^12.6.1.4"
 
 [tool.poetry.group.dev.dependencies]
 moto = "^5.0.13"