wheels: use cuda-toolkit runtime wheels

Author: jameslamb

dependencies.yaml

@@ -160,7 +160,9 @@ files:
     extras:
       table: project
     includes:
+      - cuda_wheels
       - depends_on_libraft
+      - depends_on_librmm
   py_build_pylibcugraph:
     output: pyproject
     pyproject_dir: python/pylibcugraph
@@ -262,6 +264,30 @@ dependencies:
           - matrix:
               cuda: "12.*"
             packages:
+  cuda_wheels:
+    specific:
+      - output_types: [pyproject, requirements]
+        matrices:
+          - matrix:
+              cuda: "12.*"
+              use_cuda_wheels: "true"
+            packages:
+              - cuda-toolkit[cublas,curand,cusolver,cusparse,nvjitlink]==12.*
+          - matrix:
+              cuda: "13.*"
+              use_cuda_wheels: "true"
+            packages:
+              - cuda-toolkit[cublas,curand,cusolver,cusparse,nvjitlink]==13.*
+          # if use_cuda_wheels=false is provided, do not add dependencies on any CUDA wheels
+          # (e.g. for DLFW and pip devcontainers)
+          - matrix:
+              use_cuda_wheels: "false"
+            packages:
+          # if no matching matrix selectors passed, list the unsuffixed packages
+          # (just as a source of documentation, as this populates pyproject.toml in source control)
+          - matrix:
+            packages:
+              - cuda-toolkit[cublas,curand,cusolver,cusparse,nvjitlink]>=12,<14
   common_build:
     common:
       - output_types: [conda, requirements, pyproject]

python/libcugraph/CMakeLists.txt

@@ -1,6 +1,6 @@
 # =============================================================================
 # cmake-format: off
-# SPDX-FileCopyrightText: Copyright (c) 2025, NVIDIA CORPORATION.
+# SPDX-FileCopyrightText: Copyright (c) 2025-2026, NVIDIA CORPORATION.
 # SPDX-License-Identifier: Apache-2.0
 # cmake-format: on
 # =============================================================================
@@ -42,3 +42,29 @@ set(CUGRAPH_COMPILE_CUVS ON)
 set(CUGRAPH_USE_CUVS_STATIC ON)
 
 add_subdirectory(../../cpp cugraph-cpp)
+
+# assumes libcugraph.so is installed 2 levels deep, e.g. site-packages/libcugraph/lib64/libcugraph.so
+if(CMAKE_CUDA_COMPILER_VERSION VERSION_GREATER_EQUAL 13.0)
+  # starting with CTK 13 wheels, all libraries are grouped in this 'nvidia/cu13/lib' directory
+  set(rpaths "$ORIGIN" "$ORIGIN/../../nvidia/cu13/lib")
+else()
+  set(rpaths
+      "$ORIGIN"
+      "$ORIGIN/../../nvidia/cublas/lib"
+      "$ORIGIN/../../nvidia/curand/lib"
+      "$ORIGIN/../../nvidia/cusolver/lib"
+      "$ORIGIN/../../nvidia/cusparse/lib"
+      "$ORIGIN/../../nvidia/nvjitlink/lib"
+  )
+endif()
+
+set_property(
+  TARGET cugraph
+  PROPERTY INSTALL_RPATH ${rpaths}
+  APPEND
+)
+set_property(
+  TARGET cugraph_c
+  PROPERTY INSTALL_RPATH ${rpaths}
+  APPEND
+)

python/libcugraph/pyproject.toml

@@ -26,7 +26,9 @@ classifiers = [
     "Environment :: GPU :: NVIDIA CUDA",
 ]
 dependencies = [
+    "cuda-toolkit[cublas,curand,cusolver,cusparse,nvjitlink]>=12,<14",
     "libraft==26.4.*,>=0.0.0a0",
+    "librmm==26.4.*,>=0.0.0a0",
 ] # This list was generated by `rapids-dependency-file-generator`. To make changes, edit ../../dependencies.yaml and run `rapids-dependency-file-generator`.
 
 [project.urls]