diff --git a/conda/environments/all_cuda-125_arch-x86_64.yaml b/conda/environments/all_cuda-125_arch-x86_64.yaml index 41807e89b..3c8750697 100644 --- 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 index 1ffc3b912..29135fa7f 100644 --- a/dependencies.yaml +++ b/dependencies.yaml @@ -182,6 +182,7 @@ dependencies: cuda: "12.*" packages: - cuda-cudart-dev + - cuda-nvml-dev - cuda-nvtx checks: common: