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'