前提条件

已完成PyTorch框架及混合精度模块的安装,请用户参考CANN 软件安装指南中的“安装须知”章节选择安装开发环境(Ascend-cann-toolkit)或运行环境nnae软件(Ascend-cann-nnae),并参考其中的“在昇腾设备上安装>安装深度学习框架>安装PyTorch”章节安装PyTorch框架。