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
|
|||
|
|
|