安装 目前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才能正常使用。