Add cuda-nvml-dev to dependencies.yaml.

rapidsai · Jul 23, 2024 · f79ea8a · f79ea8a
1 parent 94367fd
commit f79ea8a
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: