jaunatisblue/final-qibotn
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

Add nvidia-nccl-cu12 library to poetry.

Browse Source
This commit is contained in:
Yang Liwei
2025-11-26 18:26:55 +08:00
parent aaee0bedbd
commit e3809c7e7f
1 changed files with 2 additions and 1 deletions
Download Patch File Download Diff File Expand all files Collapse all files

3
pyproject.toml
View File

@@ -25,6 +25,7 @@ qibo = "^0.2.16"
quimb = { version = "^1.10.0", extras = ["tensor"] } quimb = { version = "^1.10.0", extras = ["tensor"] }
cupy-cuda12x = { version = "^13.6.0", optional = true } cupy-cuda12x = { version = "^13.6.0", optional = true }
cuda-toolkit = {extras = ["all"], version = "^12.9.1"} 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 } cuquantum-python-cu12 = { version = "^25.9.1", optional = true }
qmatchatea = { version = "^1.4.3", optional = true } qmatchatea = { version = "^1.4.3", optional = true }
qiskit = { version = "^1.4.0", optional = true } qiskit = { version = "^1.4.0", optional = true }
@@ -32,7 +33,7 @@ qtealeaves = { version = "^1.5.20", optional = true }
[tool.poetry.extras] [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"] qmatchatea = ["qmatchatea"]
[tool.poetry.group.docs] [tool.poetry.group.docs]