1 硬件配置
使用学校的集群,相关配置如下:
CPU:鲲鹏920
NPU:昇腾910B
操作系统:openEuler 22.03
2 安装版本
根据昇腾官方gitee上的信息,Pytoch 2.1.0是长期支持版本,因此选择安装这一版本,从而最大限度避坑。
本教程选择的版本如下:
Python:3.9
Pytorch:2.1.0
选择依据:
参考链接:https://gitee.com/ascend/pytorch
3 安装步骤
1、新建conda环境。
conda create -n pytorch python=3.9
2、在新建好的conda环境中,安装基础的依赖。
pip install attrs cython numpy==1.24.0 decorator sympy cffi pyyaml pathlib2 psutil protobuf==3.20 scipy requests absl-py --user
参考链接:
https://www.hi