1. 概述

mcPyTorch在PyTorch的基础上增加了MXMACA® 后端,支持使用MXMACA硬件加速PyTorch中各类计算任务。

当前发布包含以下版本安装包:

  • PyTorch 2.1.2 + Python 3.8

  • PyTorch 2.1.2 + Python 3.10

  • PyTorch 2.4.0 + Python 3.8

  • PyTorch 2.4.0 + Python 3.10

  • PyTorch 2.4.0 + Python 3.12

  • PyTorch 2.6.0 + Python 3.10

  • PyTorch 2.6.0 + Python 3.12

2.4.0之前版本的mcPyTorch支持x86_64/Arm架构下Ubuntu 18/20/22以及CentOS 8/9系统上运行。 2.4.0及之后版本的mcPyTorch支持x86_64/Arm架构下Ubuntu 20/22以及CentOS 8/9系统上运行。 mcPyTorch的具体支持功能,参见 3 功能支持