1. 概述

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

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

  • 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

  • PyTorch 2.8.0 + Python 3.10

  • PyTorch 2.8.0 + Python 3.12

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

1.1. 适用产品

本文档适用于沐曦MXC500系列产品:曦云® C500、C500X、C550、C550-PL、C588,曦思® N260,曦索® X206;以及MXC600系列产品:曦云® C600,曦思® N300,曦索® X301、X302。