update cuda packages in poetry

eosphoros-ai · Nov 14, 2023 · fc68110 · fc68110
1 parent 27c8e37
commit fc68110
Show file tree

Hide file tree

Showing 2 changed files with 181 additions and 1 deletion.
diff --git a/poetry.lock b/poetry.lock
diff --git a/pyproject.toml b/pyproject.toml
@@ -43,6 +43,20 @@ black = "^23.11.0"
 pyright = "^1.1.335"
 pylint = "^3.0.2"
 torch = "^2.0.0"
+markupsafe = "2.1.3"
+nvidia-cuda-nvrtc-cu12 = "12.1.105"
+nvidia-cuda-runtime-cu12 = "12.1.105"
+nvidia-cuda-cupti-cu12 = "12.1.105"
+nvidia-cudnn-cu12 = "8.9.2.26"
+nvidia-cublas-cu12 = "12.1.3.1"
+nvidia-cufft-cu12 = "11.0.2.54"
+nvidia-curand-cu12 = "10.3.2.106"
+nvidia-cusolver-cu12 = "11.4.5.107"
+nvidia-cusparse-cu12 = "12.1.0.106"
+nvidia-nccl-cu12 = "2.18.1"
+nvidia-nvtx-cu12 = "12.1.105"
+triton = "2.1.0"
+nvidia-nvjitlink-cu12 = "^12.3.52"
 
 [build-system]
 requires = ["poetry-core"]