Try including additional wheel dependencies

Signed-off-by: Ben Howe <[email protected]>
NVIDIA · Jan 10, 2025 · 68d0b41 · 68d0b41
1 parent ff7369d
commit 68d0b41
Show file tree

Hide file tree

Showing 2 changed files with 8 additions and 0 deletions.
diff --git a/libs/qec/pyproject.toml b/libs/qec/pyproject.toml
@@ -12,6 +12,10 @@ requires-python = ">=3.10"
 readme = "README.md"
 dependencies = [
   'cuda-quantum-cu12 ~= 0.9.0',
+  'nvidia-cublas-cu12 ~= 12.0',
+  'nvidia-cuda-runtime-cu12 ~= 12.0',
+  'nvidia-cusolver-cu12 ~= 11.4',
+  'nvidia-cuda-nvrtc-cu12 ~= 12.0'
 ]
 classifiers = [
     'Intended Audience :: Science/Research',

diff --git a/libs/solvers/pyproject.toml b/libs/solvers/pyproject.toml
@@ -12,6 +12,10 @@ requires-python = ">=3.10"
 readme = "README.md"
 dependencies = [
   'cuda-quantum-cu12 ~= 0.9.0',
+  'nvidia-cublas-cu12 ~= 12.0',
+  'nvidia-cuda-runtime-cu12 ~= 12.0',
+  'nvidia-cusolver-cu12 ~= 11.4',
+  'nvidia-cuda-nvrtc-cu12 ~= 12.0'
   'fastapi',
   'networkx',
   'pyscf',