安装

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

• 关于torch_npu插件的安装介绍请参考《Ascend Extension for PyTorch 配置与安装》。
• 当PyTorch版本为2.1.0，torch_npu插件版本为6.0.RC3时，才能正常使用TorchAir。
• 使用TorchAir功能时，训练/推理脚本中必须先import torch_npu，再import torchair才能正常使用。