add matx cuda dependencies

nv-morpheus · Oct 7, 2024 · 8546540 · 8546540
1 parent e041699
commit 8546540
Show file tree

Hide file tree

Showing 3 changed files with 10 additions and 2 deletions.
diff --git a/conda/environments/all_cuda-125_arch-x86_64.yaml b/conda/environments/all_cuda-125_arch-x86_64.yaml
@@ -57,8 +57,11 @@ dependencies:
 - isort
 - jsonpatch>=1.33
 - kfp
+- libcublas-dev
 - libcudf=24.10
 - libcufft-dev
+- libcurand-dev
+- libcusolver-dev
 - librdkafka>=1.9.2,<1.10.0a0
 - libtool
 - libwebp=1.3.2

diff --git a/conda/environments/dev_cuda-125_arch-x86_64.yaml b/conda/environments/dev_cuda-125_arch-x86_64.yaml
@@ -47,8 +47,11 @@ dependencies:
 - include-what-you-use=0.20
 - ipython
 - isort
+- libcublas-dev
 - libcudf=24.10
 - libcufft-dev
+- libcurand-dev
+- libcusolver-dev
 - librdkafka>=1.9.2,<1.10.0a0
 - libtool
 - libwebp=1.3.2

diff --git a/dependencies.yaml b/dependencies.yaml
@@ -214,10 +214,12 @@ dependencies:
               - cuda-nvml-dev=12.5
               - cuda-nvrtc-dev=12.5
               - cuda-nvtx-dev=12.5
-              # - cuda-tools=12.5
+              - cuda-sanitizer-api
               - cuda-version=12.5
+              - libcublas-dev # required by matx
               - libcufft-dev # required by matx
-              - cuda-sanitizer-api
+              - libcusolver-dev # required by matx
+              - libcurand-dev # required by matx