From 6e642e87b5939a37aa99b9f1c17c5748fbed5d7b Mon Sep 17 00:00:00 2001
From: =?UTF-8?q?=E9=AD=8F=E9=81=87=E5=8D=BF?= <358614401@qq.com>
Date: Fri, 24 Mar 2023 17:43:40 +0800
Subject: [PATCH] Create requirements.txt

---
 requirements.txt | 3 +++
 1 file changed, 3 insertions(+)

diff --git a/requirements.txt b/requirements.txt
index 094b32d..7a78897 100644
--- a/requirements.txt
+++ b/requirements.txt
@@ -2,10 +2,13 @@ matplotlib==3.7.0
 numpy==1.23.5
 opencv_python==4.7.0.72
 Pillow==9.4.0
+pynvml==11.5.0
 PyWavelets==1.4.1
 scikit_learn==1.2.1
 seaborn==0.12.2
 tensorflow==2.10.0
 tensorflow_intel==2.11.0
 torch==2.0.0+cu117
+torchinfo==1.7.2
+tqdm==4.64.1
 wfdb==4.1.0