diff --git a/setup.py b/setup.py index 86d29eb..13285eb 100644 --- a/setup.py +++ b/setup.py @@ -42,8 +42,6 @@ setup( "qibo>=0.1.10", "qibojit>=0.0.7", "quimb[tensor]>=1.6.0", - "cupy>=11.6.0", - "cuquantum-python-cu11", ], extras_require={ "docs": [], @@ -55,6 +53,10 @@ setup( "analysis": [ "pylint>=2.16.0", ], + "cuda": [ + "cupy>=11.6.0", + "cuquantum-python-cu11>=23.3.0", + ], }, python_requires=">=3.8.0", long_description=(HERE / "README.md").read_text(encoding="utf-8"),