安装

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

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

• 当torch_npu插件版本为7.1.0及之前版本，PyTorch版本需要为2.1.0，可正常使用TorchAir；当torch_npu插件版本为7.2.0及之后版本，均可正常使用TorchAir。
• 使用TorchAir功能时，模型脚本中必须先import torch_npu，再import torchair才能正常使用。