海光 特有的Python 包 下载地址 必须有 DCU 专用版(底层含 CUDA/ROCm 二进制)
本文针对DCU(国产AI加速卡)环境下的LLaMA-Factory大模型训练/推理需求,分类整理了关键Python包的安装策略: 必须DCU专用版:包括PyTorch、Triton、DeepSpeed、FlashAttention等含GPU二进制代码的包,必须使用海光sourcefind.cn提供的移植版本,否则会因CUDA/ROCm不兼容导致崩溃; 建议DCU优化版:如TorchVision、Apex等性能相关组件,推荐使用匹配DTK2404的版本以获得最佳性能; 纯Python通用包:如Transfor
zhangfeng11332026-10-09 20:32:12