下载
中文
注册
我要评分
文档获取效率
文档正确性
内容完整性
文档易理解
在线提单
论坛求助
昇腾小AI

安装

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

  • 关于torch_npu插件的安装介绍请参考Ascend Extension for PyTorch 配置与安装
  • 当PyTorch版本为2.1.0或2.3.1,torch_npu插件版本为6.0.RC3时,才能正常使用TorchAir。
  • 使用TorchAir功能时,训练/推理脚本中必须先import torch_npu,再import torchair才能正常使用。
搜索结果
找到“0”个结果

当前产品无相关内容

未找到相关内容,请尝试其他搜索词