From 2aa3f3e1a86893bb0d4c1cfec13ce90c43dc642a Mon Sep 17 00:00:00 2001 From: Vyas Ramasubramani Date: Wed, 3 May 2023 15:43:20 -0700 Subject: [PATCH] Update cupy dependency --- dependencies.yaml | 7 ++++++- 1 file changed, 6 insertions(+), 1 deletion(-) diff --git a/dependencies.yaml b/dependencies.yaml index 5b691d1464..630ba99c59 100644 --- a/dependencies.yaml +++ b/dependencies.yaml @@ -286,6 +286,11 @@ dependencies: common: - output_types: [conda, requirements, pyproject] packages: - - cupy - scikit-learn - scipy + - output_types: conda + packages: + - cupy>=12.0.0 + - output_types: pyproject + packages: + - cupy-cuda11x>=12.0.0