Torch兼容适配昇腾NPU
主要使用插件:torch_npu、torchvision_npu
去昇腾官方查看对应版本的cann兼容torch版本,下载对应torch,自己拉取torchvision_npu代码放置服务器上进行编译,各自编译会根据自己平台和python版本生成对应的torchvision_npu,有所区别,所以以源代码为准。
在原代码启动入口上,import这俩个包即可使用。
参考资料:
完整清单列表-第三方库支持情况-套件与三方库支持清单-Ascend Extension for PyTorch6.0.RC2开发文档-昇腾社区