From 8098317d36c044e607d1bad91193d8afbfc0984c Mon Sep 17 00:00:00 2001
From: Xiaozhe Yao <askxzyao@gmail.com>
Date: Mon, 30 Sep 2024 16:16:28 +0200
Subject: [PATCH] Update requirements.txt

---
 requirements.txt | 4 ++--
 1 file changed, 2 insertions(+), 2 deletions(-)

diff --git a/requirements.txt b/requirements.txt
index 180d275..06f1f44 100644
--- a/requirements.txt
+++ b/requirements.txt
@@ -1,3 +1,3 @@
-pynvml
+nvidia-ml-py
 rich
-ninja
\ No newline at end of file
+ninja