diff --git a/pyproject.toml b/pyproject.toml index cec1803..4c21e5a 100644 --- a/pyproject.toml +++ b/pyproject.toml @@ -50,8 +50,8 @@ 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 = [ @@ -59,8 +59,8 @@ rapids12 = [ "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"]