昇腾社区首页
中文
注册

安装

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

  • torch_npu插件的安装介绍请参考Ascend Extension for PyTorch 软件安装指南,请保证与CANN相关包的版本匹配,否则功能可能无法正常使用。
  • 当PyTorch版本为2.1.0,torch_npu插件版本为7.1.0时,才能正常使用TorchAir。
  • 使用TorchAir功能时,模型脚本中必须先import torch_npu,再import torchair才能正常使用。