昇腾社区首页
中文
注册
开发者
下载

PyTorch适配

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

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

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