Merge pull request #215 from jakirkham/use_pynvml_12

Author: jakirkham

conda/environments/all_arch-any.yaml

@@ -10,7 +10,7 @@ dependencies:
 - nodejs=18
 - pre-commit
 - psutil
-- pynvml>=11.0.0,<12.0.0a0
+- pynvml>=12.0.0,<13.0.0a0
 - pytest
 - pytest-asyncio
 - pytest-jupyter[server]>=0.6.0

conda/recipes/jupyterlab-nvdashboard/meta.yaml

@@ -26,7 +26,7 @@ requirements:
   run:
     - python >=3.8
     - jupyterlab >=4
-    - pynvml >=11.0.0,<12.0.0a0
+    - pynvml >=12.0.0,<13.0.0a0
     - psutil
 
 

dependencies.yaml

@@ -72,7 +72,7 @@ dependencies:
       - output_types: [conda, requirements, pyproject]
         packages:
           - jupyterlab>=4
-          - pynvml>=11.0.0,<12.0.0a0
+          - pynvml>=12.0.0,<13.0.0a0
           - psutil
   test_python:
     common:

jupyterlab_nvdashboard/apps/gpu.py

@@ -5,7 +5,7 @@
 
 try:
     pynvml.nvmlInit()
-except pynvml.nvml.NVMLError_LibraryNotFound:
+except pynvml.NVMLError_LibraryNotFound:
     ngpus = 0
     gpu_handles = []
 else:
@@ -29,12 +29,12 @@
             max(sum(i.value.ullVal for i in bw) * 1024**2 for bw in bandwidth)
             / 2
         )
-    except (IndexError, pynvml.nvml.NVMLError_NotSupported):
+    except (IndexError, pynvml.NVMLError_NotSupported):
         nvlink_ver = None
         max_bw = []
     try:
         pci_gen = pynvml.nvmlDeviceGetMaxPcieLinkGeneration(gpu_handles[0])
-    except (IndexError, pynvml.nvml.NVMLError_NotSupported):
+    except (IndexError, pynvml.NVMLError_NotSupported):
         pci_gen = None
 
 

pyproject.toml

@@ -31,7 +31,7 @@ classifiers = [
 dependencies = [
     "jupyterlab>=4",
     "psutil",
-    "pynvml>=11.0.0,<12.0.0a0",
+    "pynvml>=12.0.0,<13.0.0a0",
 ] # This list was generated by `rapids-dependency-file-generator`. To make changes, edit dependencies.yaml and run `rapids-dependency-file-generator`.
 dynamic = ["version", "description", "authors", "urls", "keywords"]