From e215bac930f1959095e75ebcc896c7d203ecf2f6 Mon Sep 17 00:00:00 2001
From: EuropaYou <53128780+EuropaYou@users.noreply.github.com>
Date: Fri, 1 May 2026 00:52:35 +0000
Subject: [PATCH] use nvidia-ml-py instead of pynvml

---
 requirements.txt | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

diff --git a/requirements.txt b/requirements.txt
index cb86b30..87efe35 100644
--- a/requirements.txt
+++ b/requirements.txt
@@ -2,7 +2,7 @@ deepdiff
 torch
 numpy
 Pillow
-pynvml; platform_machine != 'aarch64'
+nvidia-ml-py; platform_machine != 'aarch64'
 py-cpuinfo
 piexif
 jetson-stats; platform_machine == 'aarch64'