他的回复:
310B是支持跑pytorch的,相关信息如下:基础支持:Ascend 310B 可通过 昇腾异构计算架构 CANN(Compute Architecture for Neural Networks) 与 PyTorch 集成。具体依赖以下组件:torch_npu 插件:昇腾提供的 PyTorch 后端扩展,支持将模型部署到昇腾 NPU 设备16。CANN 工具链:包含算子库、编译器(ATC)、运行时库等,实现 PyTorch 模型的编译与执行2。版本要求:PyTorch 2.1 及以上:原生支持昇腾后端,通过 torch_npu 插件直接调用昇腾设备1。PyTorch 2.1 以下:需手动迁移代码,将 CUDA 相关操作替换为 NPU 接口(如 torch.cuda → torch.npu),并通过昇腾提供的 自动迁移工具 简化适配12。但是310B主要是用于推理场景的,算力可能无法支撑训练场景,需要模型训练的话,建议还是用昇腾910系列的芯片。