From 77abf4e98a7285488cb3d4082db0d3c8269b6fbc Mon Sep 17 00:00:00 2001
From: SWivid <swivid@qq.com>
Date: Sun, 13 Oct 2024 09:46:03 +0800
Subject: [PATCH] separate torch pkg install

---
 README.md        | 7 +++++++
 requirements.txt | 4 ++--
 2 files changed, 9 insertions(+), 2 deletions(-)

diff --git a/README.md b/README.md
index 5a5b6b8..cfefbc3 100644
--- a/README.md
+++ b/README.md
@@ -25,6 +25,13 @@ Install packages:
 pip install -r requirements.txt
 ```
 
+Install torch with your CUDA version, e.g. :
+
+```bash
+pip install torch==2.3.0+cu118 --extra-index-url https://download.pytorch.org/whl/cu118
+pip install torchaudio==2.3.0+cu118 --extra-index-url https://download.pytorch.org/whl/cu118
+```
+
 ## Prepare Dataset
 
 Example data processing scripts for Emilia and Wenetspeech4TTS, and you may tailor your own one along with a Dataset class in `model/dataset.py`.
diff --git a/requirements.txt b/requirements.txt
index 9be7f3b..b88c7ff 100644
--- a/requirements.txt
+++ b/requirements.txt
@@ -10,8 +10,8 @@ jiwer
 librosa
 matplotlib
 pypinyin
-torch>=2.0
-torchaudio>=2.3.0
+# torch>=2.0
+# torchaudio>=2.3.0
 torchdiffeq
 tqdm>=4.65.0
 transformers