Move to cuda_core 1.0.0

mdboom · mdboom · commit eac40b77cb70 · 2026-05-19T09:29:58.000-04:00
diff --git a/conda/environments/all_arch-any.yaml b/conda/environments/all_arch-any.yaml
@@ -6,7 +6,7 @@ channels:
 - conda-forge
 dependencies:
 - cuda-bindings>=12.9.6,!=13.0.*,!=13.1.*
-- cuda-core @ git+https://github.com/nvidia/cuda-python@main#subdirectory=cuda_core
+- cuda-core>=1.0.0
 - jupyterlab>=4.0.0
 - nodejs=18
 - pre-commit
diff --git a/dependencies.yaml b/dependencies.yaml
@@ -74,7 +74,7 @@ dependencies:
           - jupyterlab>=4.0.0
           - psutil
           - cuda-bindings>=12.9.6,!=13.0.*,!=13.1.*
-          - cuda-core @ git+https://github.com/nvidia/cuda-python@main#subdirectory=cuda_core
+          - cuda-core>=1.0.0
   test_python:
     common:
       - output_types: [conda, requirements, pyproject]
diff --git a/jupyterlab_nvdashboard/apps/gpu.py b/jupyterlab_nvdashboard/apps/gpu.py
@@ -181,21 +181,8 @@ def send_data(self):
         # Max PCIe Throughput = BW-per-lane * Width
         max_rxtx_tp = pci_width * pci_bw[pci_gen]
 
-        pci_tx = [
-            device.pci_info.get_throughput(
-                system.PcieUtilCounter.PCIE_UTIL_TX_BYTES
-            )
-            * 1024
-            for device in gpus
-        ]
-
-        pci_rx = [
-            device.pci_info.get_throughput(
-                system.PcieUtilCounter.PCIE_UTIL_RX_BYTES
-            )
-            * 1024
-            for device in gpus
-        ]
+        pci_tx = [device.pci_info.tx_throughput * 1024 for device in gpus]
+        pci_rx = [device.pci_info.rx_throughput * 1024 for device in gpus]
 
         stats = {
             "pci_tx": pci_tx,
diff --git a/pyproject.toml b/pyproject.toml
@@ -31,7 +31,7 @@ classifiers = [
 ]
 dependencies = [
     "cuda-bindings>=12.9.6,!=13.0.*,!=13.1.*",
-    "cuda-core @ git+https://github.com/nvidia/cuda-python@main#subdirectory=cuda_core",
+    "cuda-core>=1.0.0",
     "jupyterlab>=4.0.0",
     "psutil",
 ] # This list was generated by `rapids-dependency-file-generator`. To make changes, edit dependencies.yaml and run `rapids-dependency-file-generator`.

Original file line number	Diff line number	Diff line change
`@@ -31,7 +31,7 @@ classifiers = [`
`31`	`31`	`]`
`32`	`32`	`dependencies = [`
`33`	`33`	`"cuda-bindings>=12.9.6,!=13.0.,!=13.1.",`
`34`		`- "cuda-core @ git+https://github.com/nvidia/cuda-python@main#subdirectory=cuda_core",`
	`34`	`+ "cuda-core>=1.0.0",`
`35`	`35`	`"jupyterlab>=4.0.0",`
`36`	`36`	`"psutil",`
`37`	`37`	] # This list was generated by `rapids-dependency-file-generator`. To make changes, edit dependencies.yaml and run `rapids-dependency-file-generator`.