32 lines
525 B
Plaintext
Executable File
32 lines
525 B
Plaintext
Executable File
# 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
|
||
|