From 5507030dcac8cdaee66f2fc6a083b765d8d9a8e1 Mon Sep 17 00:00:00 2001 From: Sebastian Berg Date: Fri, 23 Aug 2024 21:34:33 +0200 Subject: [PATCH] Add numpy and torch as runtime requirements --- conda/environments/all_cuda-118_arch-x86_64.yaml | 1 + conda/environments/all_cuda-125_arch-x86_64.yaml | 1 + dependencies.yaml | 10 ++++++++-- 3 files changed, 10 insertions(+), 2 deletions(-) diff --git a/conda/environments/all_cuda-118_arch-x86_64.yaml b/conda/environments/all_cuda-118_arch-x86_64.yaml index 24f99ab23..3af2f9da2 100644 --- 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 index 2c528e087..bd7b16d60 100644 --- 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 index 756f93076..ec74b7235 100644 --- 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]