| # 核心框架 | |
| torch==2.4.0 # 保持与conda环境一致（pypi安装） | |
| torchvision==0.19.0 # 保持与conda环境一致（pypi安装） | |
| # Hugging Face 生态 | |
| transformers==4.44.0 # 原4.39.1 → conda实际安装4.44.0 | |
| diffusers==0.31.0 # 保持与conda环境一致 | |
| accelerate==0.34.2 # 原0.27.2 → conda实际安装0.34.2 | |
| peft==0.12.0 # 原git提交 → conda实际安装0.12.0（pypi） | |
| datasets==2.20.0 # 保持与conda环境一致 | |
| # 工具链 | |
| deepspeed==0.15.4 # 原0.14.2 → conda实际安装0.15.4 | |
| # bitsandbytes==0.44.1 # 原0.43.0 → conda实际安装0.44.1 | |
| protobuf==3.20.0 # 原3.20.3 → conda实际安装3.20.0（需验证tensorboard兼容性） | |
| tensorboard==2.18.0 # 新增明确版本（conda实际安装2.18.0） | |
| tensorboardx==2.6.2.2 # 新增明确版本（conda实际安装2.6.2.2） | |
| webdataset==0.2.100 # 新增明确版本（conda实际安装0.2.100） | |
| # 训练辅助 | |
| warmup_scheduler==0.3 # 新增明确版本（conda实际安装0.3） | |
| torchmetrics==1.6.0 # 新增明确版本（conda实际安装1.6.0） | |
| open_clip_torch==2.29.0 # 新增明确版本（conda实际安装2.29.0） | |
| evaluate==0.4.3 # 新增明确版本（conda实际安装0.4.3） | |
| bert_score==0.3.13 # 新增明确版本（conda实际安装0.3.13） | |
| einops==0.8.0 # 保持与conda环境一致 | |
| wandb==0.17.7 # 保持与conda环境一致 | |
| # 图像处理 | |
| matplotlib==3.9.2 # 新增明确版本（conda实际安装3.9.2） | |
| opencv-python==4.10.0.84 # 新增明确版本（conda实际安装4.10.0.84） | |
| clean-fid==0.1.35 # 新增明确版本（conda实际安装0.1.35） | |
| skia-python==87.6 # 新增明确版本（conda实际安装87.6） | |
| # 部署与接口 | |
| # gradio==5.5.0 # 新增明确版本（conda实际安装5.5.0） | |
| langchain>=0.0.139 # 保持约束（conda实际安装0.3.7符合要求） | |
| tiktoken==0.8.0 # 新增明确版本（conda实际安装0.8.0） | |
| # 系统工具 | |
| ninja==1.11.1.1 # 新增明确版本（conda实际安装1.11.1.1） | |
| pynvml==11.5.3 # 新增明确版本（conda实际安装11.5.3） | |
| colorama==0.4.6 # 新增明确版本（conda实际安装0.4.6） | |
| click>=8.0.4,<9 # 保持约束（conda实际安装8.1.7符合要求）\ | |
| sentencepiece | |
| # random |