From f63310c2f4d538c2872de1b355c1735f6de87406 Mon Sep 17 00:00:00 2001 From: Bradley Dice Date: Mon, 12 Feb 2024 11:46:38 -0600 Subject: [PATCH] Add cuda-nvtx-dev, add missing CUDA library dependencies. (#4162) This PR fixes issues with devcontainer builds where `cuda-nvtx-dev` was missing when building `libcugraph_etl`. Authors: - Bradley Dice (https://github.com/bdice) Approvers: - Jake Awe (https://github.com/AyodeAwe) URL: https://github.com/rapidsai/cugraph/pull/4162 --- conda/environments/all_cuda-118_arch-x86_64.yaml | 1 + conda/environments/all_cuda-120_arch-x86_64.yaml | 7 +++++++ dependencies.yaml | 10 +++++++++- 3 files changed, 17 insertions(+), 1 deletion(-) diff --git a/conda/environments/all_cuda-118_arch-x86_64.yaml b/conda/environments/all_cuda-118_arch-x86_64.yaml index 43bc60d91fb..de8db94df1c 100644 --- a/conda/environments/all_cuda-118_arch-x86_64.yaml +++ b/conda/environments/all_cuda-118_arch-x86_64.yaml @@ -14,6 +14,7 @@ dependencies: - breathe - c-compiler - cmake>=3.26.4 +- cuda-nvtx - cuda-version=11.8 - cudatoolkit - cudf==24.4.* diff --git a/conda/environments/all_cuda-120_arch-x86_64.yaml b/conda/environments/all_cuda-120_arch-x86_64.yaml index 1829f3dd860..03dade0ed1f 100644 --- a/conda/environments/all_cuda-120_arch-x86_64.yaml +++ b/conda/environments/all_cuda-120_arch-x86_64.yaml @@ -14,7 +14,10 @@ dependencies: - breathe - c-compiler - cmake>=3.26.4 +- cuda-cudart-dev - cuda-nvcc +- cuda-nvtx-dev +- cuda-profiler-api - cuda-version=12.0 - cudf==24.4.* - cupy>=12.0.0 @@ -29,8 +32,12 @@ dependencies: - graphviz - gtest>=1.13.0 - ipython +- libcublas-dev - libcudf==24.4.* - libcugraphops==24.4.* +- libcurand-dev +- libcusolver-dev +- libcusparse-dev - libraft-headers==24.4.* - libraft==24.4.* - librmm==24.4.* diff --git a/dependencies.yaml b/dependencies.yaml index cfefe3b9ff9..58354407bbc 100644 --- a/dependencies.yaml +++ b/dependencies.yaml @@ -327,10 +327,18 @@ dependencies: - matrix: cuda: "12.*" packages: + - cuda-cudart-dev + - cuda-nvtx-dev + - cuda-profiler-api + - libcublas-dev + - libcurand-dev + - libcusolver-dev + - libcusparse-dev - matrix: cuda: "11.*" packages: - cudatoolkit + - cuda-nvtx common_build: common: - output_types: [conda, pyproject] @@ -345,6 +353,7 @@ dependencies: - cxx-compiler - gmock>=1.13.0 - gtest>=1.13.0 + - libcudf==24.4.* - libcugraphops==24.4.* - libraft-headers==24.4.* - libraft==24.4.* @@ -438,7 +447,6 @@ dependencies: packages: - aiohttp - fsspec>=0.6.0 - - libcudf==24.4.* - requests - nccl>=2.9.9 - ucx-proc=*=gpu