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 功能支持。