安装

目前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版本请参见对应版本文档中的安装介绍。
• 为确保正常使用TorchAir功能，PyTorch建议使用2.6.0及以上版本。