From b84ccadb208143d558d0cfd2483eedb763f3e581 Mon Sep 17 00:00:00 2001 From: Liwei Yang Date: Wed, 30 Aug 2023 16:41:17 +0800 Subject: [PATCH] Move cupy package to extras to avoid the issue of missing CUDA --- setup.py | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) diff --git a/setup.py b/setup.py index 38c647c..e781945 100644 --- a/setup.py +++ b/setup.py @@ -42,7 +42,6 @@ setup( "qibo>=0.1.10", "qibojit>=0.0.7", "quimb[tensor]>=1.4.0", - "cupy>=11.6.0", ], extras_require={ "docs": [], @@ -54,7 +53,8 @@ setup( "analysis": [ "pylint>=2.16.0", ], - "cuquantum": [ + "cuda": [ + "cupy>=11.6.0", "cuquantum-python-cu11>=23.3.0", ], },