From f5de1ab3cb33856f3c8fe79fb621b0b74c446ae1 Mon Sep 17 00:00:00 2001 From: =?UTF-8?q?Tobias=20H=C3=B6lzer?= Date: Fri, 29 Nov 2024 19:01:03 +0100 Subject: [PATCH] Remove source-cupy dependency --- pyproject.toml | 8 ++++---- 1 file changed, 4 insertions(+), 4 deletions(-) diff --git a/pyproject.toml b/pyproject.toml index b252aee..eb9829c 100644 --- a/pyproject.toml +++ b/pyproject.toml @@ -60,8 +60,8 @@ cuda11 = [ "torch==2.2.0+cu118", "torchvision==0.17.0+cu118", "cucim-cu11>=24.8.0", - "cupy>=13.3.0", - "cupy-xarray>=0.1.4", + "cupy-cuda11x>=13.3.0", + "cupy-xarray @ git+https://github.com/xarray-contrib/cupy-xarray", "cuda-python>=12.6.2.post1", "pynvml==11.4.1", ] @@ -69,8 +69,8 @@ cuda12 = [ "torch==2.2.0+cu121", "torchvision==0.17.0+cu121", "cucim-cu12==24.8.*", - "cupy>=13.3.0", - "cupy-xarray>=0.1.4", + "cupy-cuda12x>=13.3.0", + "cupy-xarray @ git+https://github.com/xarray-contrib/cupy-xarray", "cuda-python>=12.6.2.post1", "pynvml==11.4.1", ]