昇腾平台提供了PyTorch适配插件Ascend Extension for PyTorch,使得原生PyTorch模型可以方便的在昇腾平台上执行。
Ascend Extension for PyTorch支持PyTorch 2.2.0、2.1.0与1.11.0版本,开发者可以参见《Ascend Extension for PyTorch API 参考》查询Ascend Extension for PyTorch对原生PyTorch框架的API支持情况,以及Ascend Extension for PyTorch自定义API的功能描述、参数说明与使用示例等。