增加混电算例功能

2026-04-07 19:23:12 +08:00
parent 57ba7b5fee
commit cf9a1a2654
16 changed files with 4338 additions and 838 deletions
@@ -1,7 +1,7 @@
 # requirements.txt

-# ===== 新增：PyTorch CPU 轮子下载源 =====
--extra-index-url https://download.pytorch.org/whl/cpu
+# ===== PyTorch：GPU (CUDA 12.1) 版本，如无 NVIDIA GPU 可改为 cpu =====
+--extra-index-url https://download.pytorch.org/whl/cu121

 gradio==4.44.1
 gradio-client==1.3.0
@@ -14,10 +14,11 @@ matplotlib==3.9.4
 aiohttp==3.13.5
 pillow==10.4.0
 # ===== 新增：混动模型（Model）运行依赖 =====
-torch==2.4.1
+torch==2.4.1+cu121
 botorch==0.14.0
 gpytorch==1.14
 pyro-ppl==1.9.1
 pandas==2.3.3
-scipy==1.15.3
+scipy>=1.10
 scikit-learn==1.7.1
+psutil>=5.9