fixed installation

EliHei2 · Sep 23, 2024 · a458290 · a458290
1 parent 0863be2
commit a458290
Showing 1 changed file with 4 additions and 4 deletions.
diff --git a/pyproject.toml b/pyproject.toml
@@ -50,17 +50,17 @@ rapids11 = [
     "cudf-cu11==24.8.*",
     "cugraph-cu11==24.8.*",
     "cuvs-cu11==24.8.*",
-    "nccl==2.12.*",
-    "cutensor==1.3.*"
+    "nvidia-nccl-cu11==2.18.*",
+    "cutensor-cu11"
 ]
 
 rapids12 = [
     "cupy-cuda12x",
     "cudf-cu12==24.8.*",
     "cugraph-cu12==24.8.*",
     "cuvs-cu12==24.8.*",
-    "nccl==2.13.*",
-    "cutensor==1.4.*"
+    "nvidia-nccl-cu12==2.18.*",
+    "cutensor-cu12"
 ]
 
 multiprocessing = ["multiprocessing"]