文档
注册

概述

提供基于原生PyTorch框架2.1/2.0/1.11.0的API接口支持情况与限制说明以及PyTorch Adapter自定义API的接口说明、功能描述、参数说明与使用示例。

昇腾PyTorch提供的自定义接口参考PyTorch社区公开接口规范:Public API definition and documentation,与官方提示一致。如果一个函数符合公开接口的标准且在文档中有展示,则该接口是公开接口。否则,使用该功能前可以在社区询问该功能是否确实是公开的或意外暴露的接口,因为这些未暴露接口将来可能会被修改或者删除。

昇腾PyTorch使用monkey-patch方式实现PyTorch接口的适配,通过将昇腾PyTorch中的部分接口动态替换PyTorch中的接口实现对部分接口的支持,推荐用户使用文档中推荐的调用方式方式。

昇腾PyTorch项目采用C++和PyThon联合开发,当前除Libtorch场景外正式接口只提供Python接口,在torch_npu的二进制包中动态库不直接提供服务,暴露的接口为内部使用,不建议用户使用。

PyTorch Adapter不支持单进程多卡。

搜索结果
找到“0”个结果

当前产品无相关内容

未找到相关内容,请尝试其他搜索词