# 核心框架 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