Create requirements.txt

requirements.txt

@@ -0,0 +1,54 @@
+# ========================
+# 核心框架 
+# ========================
+torch>=2.3.0  # 基础框架 
+torchvision>=0.18.0 
+pytorch-lightning>=2.2.0  # 训练管理 
+
+# ========================
+# 模型压缩专用库 
+# ========================
+model-compression-toolkit>=1.8.0  # 量化/剪枝工具包 
+nni>=3.0.0  # 微软神经网络智能压缩组件 
+openvino-dev==2025.1.0  # Intel模型优化工具 
+
+# ========================
+# 计算优化 
+# ========================
+thop>=0.2.0  # FLOPs计算 
+onnxruntime-gpu>=1.17.0  # 加速推理 
+tensorrt==8.6.1.post5   # NVIDIA部署优化 
+
+# ========================
+# 辅助工具 
+# ========================
+wandb>=0.18.0  # 实验跟踪 
+mlflow>=3.0.0  # 模型管理 
+networkx>=3.2  # 网络结构分析 
+
+# ========================
+# 数据处理 
+# ========================
+numpy>=1.26.0 
+pandas>=2.2.0 
+scipy>=1.13.0 
+
+# ========================
+# 可视化 
+# ========================
+matplotlib>=3.8.0 
+seaborn>=0.14.0 
+plotly>=5.18.0 
+
+# ========================
+# 部署支持 
+# ========================
+onnx>=1.15.0 
+openvino>=2025.1.0 
+tvm>=1.2.0  # 跨平台编译 
+
+# ========================
+# 可选组件（按需安装）
+# ========================
+# apex @ git+https://github.com/NVIDIA/apex   # 混合精度训练 
+# intel-extension-for-pytorch>=2.3.10  # Intel硬件加速