昇腾社区首页
中文
注册

环境准备

安装CANN软件包

请用户参照CANN 软件安装指南的“安装开发环境”章节,依据设备情况及操作系统支持情况,安装CANN软件包。

安装PyTorch

MindIE中各组件依赖Pytorch框架、Torch-NPU插件,依赖情况如下表所示,请用户依据实际使用需求安装。

表1 MindIE下部分组件依赖Torch情况说明表

组件名称

是否需要安装Pytorch框架

是否需要安装Torch-NPU插件

MindIE-Torch

必装

可选

MindIE-Service

必装

必装