Add cuda-nvml-dev to dependencies.yaml. (#197)

rapidsai · Jul 23, 2024 · 563ed8c · 563ed8c
1 parent 94367fd
commit 563ed8c
Show file tree

Hide file tree

Showing 2 changed files with 2 additions and 0 deletions.
diff --git a/conda/environments/all_cuda-125_arch-x86_64.yaml b/conda/environments/all_cuda-125_arch-x86_64.yaml
@@ -14,6 +14,7 @@ dependencies:
 - cmake>=3.26.4,!=3.30.0
 - cuda-cudart-dev
 - cuda-nvcc
+- cuda-nvml-dev
 - cuda-nvtx
 - cuda-version=12.5
 - cudnn=8.8

diff --git a/dependencies.yaml b/dependencies.yaml
@@ -182,6 +182,7 @@ dependencies:
               cuda: "12.*"
             packages:
               - cuda-cudart-dev
+              - cuda-nvml-dev
               - cuda-nvtx
   checks:
     common: