Add more cuda depencencies for some reason...

biaslyze-dev · Jun 1, 2023 · a1907a1 · a1907a1
1 parent 805f700
commit a1907a1
Show file tree

Hide file tree

Showing 2 changed files with 219 additions and 1 deletion.
diff --git a/poetry.lock b/poetry.lock
diff --git a/pyproject.toml b/pyproject.toml
@@ -35,6 +35,18 @@ spacy = "^3.5.0"
 transformers = "^4.26.1"
 bertopic = "0.13.0"
 umap-learn = "^0.5.3"
+nvidia-cublas-cu11 = "^11.11.3.6"
+nvidia-cudnn-cu11 = "^8.9.1.23"
+nvidia-cuda-cupti-cu11 = "^11.8.87"
+nvidia-cuda-nvrtc-cu11 = "^11.8.89"
+nvidia-cuda-runtime-cu11 = "^11.8.89"
+nvidia-cufft-cu11 = "^10.9.0.58"
+nvidia-curand-cu11 = "^10.3.0.86"
+nvidia-cusolver-cu11 = "^11.4.1.48"
+nvidia-cusparse-cu11 = "^11.7.5.86"
+nvidia-nccl-cu11 = "^2.16.5"
+nvidia-nvtx-cu11 = "^11.8.86"
+triton = "^2.0.0.post1"
 
 [tool.poetry.group.dev.dependencies]
 mkdocs-material = "^9.1.1"