Torch兼容适配昇腾NPU

327 阅读1分钟

Torch兼容适配昇腾NPU

主要使用插件:torch_npu、torchvision_npu

去昇腾官方查看对应版本的cann兼容torch版本,下载对应torch,自己拉取torchvision_npu代码放置服务器上进行编译,各自编译会根据自己平台和python版本生成对应的torchvision_npu,有所区别,所以以源代码为准。

在原代码启动入口上,import这俩个包即可使用。

参考资料:

完整清单列表-第三方库支持情况-套件与三方库支持清单-Ascend Extension for PyTorch6.0.RC2开发文档-昇腾社区

Ascend/vision

image.png

image 1.png