昇腾社区首页
中文
注册
开发者
下载

安装

目前TorchAir暂未提供独立软件包,而是作为Ascend Extension for PyTorch的三方库,随着torch_npu包一起发布。请直接安装torch_npu插件,即可使用TorchAir。

torch_npu的安装操作具体参考Ascend Extension for PyTorch 软件安装指南,请保证与CANN相关包的版本匹配(参见版本说明),否则功能可能无法正常使用。

需要注意的是,当安装的torch_npu版本为7.3.0及之后版本,均可正常使用TorchAir,对于其他torch_npu版本请参见对应版本文档中的安装介绍。