昇腾社区首页
中文
注册

PyTorch适配

MindStudio IDE暂不支持PyTorch算子适配,在完成TBE或AI CPU算子开发后,PyTorch适配请参见PyTorch算子开发指南

PyTorch适配请读者根据PyTorch算子开发指南完成以下操作。

  • 安装PyTorch依赖环境
  • 编译和安装PyTorch框架
  • PyTorch框架下适配TBE或AI CPU算子