From 41b75ef508a0a48b1cd73b6617f03aa9ba7d0079 Mon Sep 17 00:00:00 2001
From: houboxue <47343901+Albertchamberlain@users.noreply.github.com>
Date: Mon, 10 Feb 2025 23:13:17 +0800
Subject: [PATCH] Create requirements.txt

add requirements
---
 requirements.txt | 54 ++++++++++++++++++++++++++++++++++++++++++++++++
 1 file changed, 54 insertions(+)
 create mode 100644 requirements.txt

diff --git a/requirements.txt b/requirements.txt
new file mode 100644
index 0000000..29b2d4e
--- /dev/null
+++ b/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硬件加速