环境准备

请参考《CANN 软件安装指南》安装CANN软件，并参考其中的“安装开发环境>在昇腾设备上安装>安装深度学习框架>安装PyTorch”章节安装PyTorch框架。

硬件支持

表1 aarch64架构推荐硬件表
硬件名称	硬件规格
服务器	Atlas 800 训练服务器（型号9000）
CPU	鲲鹏920处理器
昇腾AI处理器	昇腾910 AI处理器
硬盘	NVMe 3.2T
内存	750GRAM

表2 X86_64架构推荐硬件表
硬件名称	硬件规格
服务器	Atlas 800 训练服务器（型号9010）
CPU	Intel Platinum 8260处理器
昇腾AI处理器	昇腾910 AI处理器
硬盘	NVMe 3.2T
内存	750GRAM