Add numpy and torch as runtime requirements

rapidsai · Aug 23, 2024 · 5507030 · 5507030
1 parent 2a33517
commit 5507030
Show file tree

Hide file tree

Showing 3 changed files with 10 additions and 2 deletions.
diff --git a/conda/environments/all_cuda-118_arch-x86_64.yaml b/conda/environments/all_cuda-118_arch-x86_64.yaml
@@ -41,6 +41,7 @@ dependencies:
 - python>=3.9,<3.12
 - pytorch-cuda=11.8
 - pytorch=2.0.0
+- pytorch>=2.0
 - rapids-build-backend>=0.3.0,<0.4.0.dev0
 - recommonmark
 - scikit-build-core>=0.10.0

diff --git a/conda/environments/all_cuda-125_arch-x86_64.yaml b/conda/environments/all_cuda-125_arch-x86_64.yaml
@@ -40,6 +40,7 @@ dependencies:
 - pytest-forked
 - pytest-xdist
 - python>=3.9,<3.12
+- pytorch>=2.0
 - rapids-build-backend>=0.3.0,<0.4.0.dev0
 - recommonmark
 - scikit-build-core>=0.10.0

diff --git a/dependencies.yaml b/dependencies.yaml
@@ -210,8 +210,14 @@ dependencies:
               - python>=3.9,<3.12
   run:
     common:
-      - output_types: [conda, requirements]
-        packages: []
+      - output_types: [conda]
+        packages:
+          - numpy>=1.23,<3.0a0
+          - pytorch>=2.0
+      - output_types: [requirements, pyproject]
+        packages:
+          - numpy>=1.23,<3.0a0
+          - torch>=2.0
   test_cpp:
     common:
       - output_types: [conda]