Add mirrors for torch (#857)

* Add mirrors for torch * set pypi mirror to bfsu, remove version pin --------- Co-authored-by: Stardust·减 <[email protected]>
fishaudio · Jan 26, 2025 · 8ae4d19 · 8ae4d19
1 parent 91bf150
commit 8ae4d19
Showing 1 changed file with 5 additions and 1 deletion.
diff --git a/install_env.bat b/install_env.bat
@@ -133,7 +133,11 @@ if "%USE_MIRROR%"=="true" (
 echo "HF_ENDPOINT: !HF_ENDPOINT!"
 echo "NO_PROXY: !no_proxy!"
 
-%PIP_CMD% install torch==2.4.1 torchvision==0.19.1 torchaudio==2.4.1 --index-url https://download.pytorch.org/whl/cu121
+if "!USE_MIRROR!" == "true" (
+    %PIP_CMD% install torch torchvision torchaudio -U --extra-index-url https://mirrors.bfsu.edu.cn/pypi/web/simple
+) else (
+    %PIP_CMD% install torch torchvision torchaudio -U --index-url https://download.pytorch.org/whl/cu121
+)
 
 %PIP_CMD% install -e . --upgrade-strategy only-if-needed