Move cupy package to extras to avoid the issue of missing CUDA

2023-08-30 16:41:17 +08:00
parent 4902195902
commit b84ccadb20
1 changed files with 2 additions and 2 deletions
--- 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",
        ],
    },