32 lines
525 B
Plaintext
32 lines
525 B
Plaintext
|
|
# AICAS 2026 - 环境依赖
|
|||
|
|
# ============================================
|
|||
|
|
|
|||
|
|
# 核心框架
|
|||
|
|
torch>=2.0.0
|
|||
|
|
transformers>=4.40.0
|
|||
|
|
accelerate>=0.25.0
|
|||
|
|
|
|||
|
|
# 数据处理
|
|||
|
|
datasets>=2.14.0
|
|||
|
|
Pillow>=9.0.0
|
|||
|
|
|
|||
|
|
# 进度条
|
|||
|
|
tqdm>=4.65.0
|
|||
|
|
|
|||
|
|
# 系统信息(可选,用于获取详细的硬件信息)
|
|||
|
|
psutil>=5.9.0
|
|||
|
|
|
|||
|
|
# 可选:Triton 算子开发
|
|||
|
|
triton>=2.1.0
|
|||
|
|
|
|||
|
|
# 可选:Flash Attention(需要 CUDA 编译)
|
|||
|
|
# flash-attn>=2.0.0
|
|||
|
|
|
|||
|
|
# 可选:量化工具
|
|||
|
|
# bitsandbytes>=0.41.0
|
|||
|
|
# auto-gptq>=0.5.0
|
|||
|
|
|
|||
|
|
# 可选:Profiling
|
|||
|
|
# tensorboard>=2.14.0
|
|||
|
|
|