Merge pull request #27 from Imageomics/fix/pynvml-deprecation

NetZissou · web-flow · commit e0236778eca9 · 2026-03-30T17:29:39.000-04:00
Replace deprecated pynvml with nvidia-ml-py
diff --git a/pyproject.toml b/pyproject.toml
@@ -45,6 +45,7 @@ dependencies = [
     "numpy>=1.21.0",           # Basic numpy operations
     "matplotlib>=3.5.0",       # Profiling plots and visualizations
     "psutil>=5.9.0",           # System resource monitoring
+    "nvidia-ml-py>=11.5.0",    # GPU monitoring and profiling
     "tqdm>=4.64.0",            # Progress bars for profiling
 ]
 

Original file line number	Diff line number	Diff line change
`@@ -45,6 +45,7 @@ dependencies = [`
`45`	`45`	`"numpy>=1.21.0", # Basic numpy operations`
`46`	`46`	`"matplotlib>=3.5.0", # Profiling plots and visualizations`
`47`	`47`	`"psutil>=5.9.0", # System resource monitoring`
	`48`	`+ "nvidia-ml-py>=11.5.0", # GPU monitoring and profiling`
`48`	`49`	`"tqdm>=4.64.0", # Progress bars for profiling`
`49`	`50`	`]`
`50`	`51`