diff --git a/pyproject.toml b/pyproject.toml index 65524c0..01a8630 100644 --- a/pyproject.toml +++ b/pyproject.toml @@ -25,6 +25,7 @@ qibo = "^0.2.16" quimb = { version = "^1.10.0", extras = ["tensor"] } cupy-cuda12x = { version = "^13.6.0", optional = true } cuda-toolkit = {extras = ["all"], version = "^12.9.1"} +nvidia-nccl-cu12 = { version = "^2.16.5", optional = true } cuquantum-python-cu12 = { version = "^25.9.1", optional = true } qmatchatea = { version = "^1.4.3", optional = true } qiskit = { version = "^1.4.0", optional = true } @@ -32,7 +33,7 @@ qtealeaves = { version = "^1.5.20", optional = true } [tool.poetry.extras] -cuda = ["cupy-cuda12x", "cuda-toolkit", "cuquantum-python-cu12", "mpi4py"] +cuda = ["cupy-cuda12x", "cuda-toolkit", "nvidia-nccl-cu12", "cuquantum-python-cu12", "mpi4py"] qmatchatea = ["qmatchatea"] [tool.poetry.group.docs]