build: Downgrade CuPy to CUDA 11, to sync with cuQuantum

This commit is contained in:
Alessandro Candido
2024-03-02 10:56:25 +01:00
parent f4311331b2
commit a5c24a66b0
2 changed files with 15 additions and 15 deletions

View File

@@ -23,12 +23,12 @@ packages = [{ include = "qibotn", from = "src" }]
python = "^3.9,<3.12"
qibo = "^0.2.4"
quimb = { version = "^1.6.0", extras = ["tensor"] }
cupy-cuda12x = { version = "^11.6.0", optional = true }
cupy-cuda11x = { version = "^11.6.0", optional = true }
cuquantum-python-cu11 = { version = "^23.3.0", optional = true }
mpi4py = { version = "^3.1.5", optional = true }
[tool.poetry.extras]
cuda = ["cupy-cuda12x", "cuquantum-python-cu11", "mpi4py"]
cuda = ["cupy-cuda11x", "cuquantum-python-cu11", "mpi4py"]
[tool.poetry.group.docs]
optional = true