欢迎使用MindTorch

_images/MindTorch_LOGO.png

文档版本: dev

MindTorch将PyTorch训练脚本高效迁移至MindSpore框架执行的工具,其目的是在不改变原有PyTorch用户的使用习惯情况下,使得PyTorch代码能在昇腾上获得高效性能。

_images/MSA_F.png
  • PyTorch接口支持: MindTorch目前支持大部分PyTorch常用接口适配。用户接口使用方式不变,基于MindSpore动态图或静态图模式下执行在昇腾算力平台上。可以在[torch接口支持列表](SupportedList.md)中查看接口支持情况。

  • TorchVision接口支持: MindTorch TorchVision是迁移自PyTorch官方实现的计算机视觉工具库,延用PyTorch官方API设计与使用习惯,内部计算调用MindSpore算子,实现与torchvision原始库同等功能。可以在[TorchVision接口支持列表](TorchVision_SupportedList.md)中查看接口支持情况。

备注

如果你在线阅读文档时遇到问题,可以在`MSAdapter`上下载仓库, 然后转到`readthedocs/build_zh/html/index.html`来离线阅读文档。 build_zh``文件夹可以在/doc/readthedocs/``中使用``make html``生成.

用户指南

MindTorch用户指南解释了如何安装MindSpore,MindTorch、版本配套、使用MindTorch例子,以及如何作为开发人员为库做出贡献。

入门

MindTorch入门指南介绍了如何使用MindTorch,微分接口、优化器和学习率、分布式适配,混合精度训练,以及模型保存和加载。

进阶

MindTorch进阶指南介绍了如何使用MindTorch进行静态图加速以及结合MindSpore框架并行训练。

MindTorch调试调优介绍了如何使用MindTorch进行精度调优和性能调优。

API

MindTorch API详细列出了和PyTorch API功能的对比。