1、安装包位置:
/data/ai_install_packages
2、添加HwHiAiUser用户:
groupadd -g 1000 HwHiAiUser
useradd -g HwHiAiUser -u 1000 -d /home/HwHiAiUser -m HwHiAiUser -s /bin/bash
3、安装驱动:
./Ascend-hdk-310p-npu-driver_6.0.0_linux-aarch64.run --full
4、安装固件:
./Ascend-hdk-310p-npu-firmware_6.0.0.run --install
5、安装cann:
./Ascend-cann-toolkit_6.3.RC2.alpha001_linux-aarch64.run --full
add source /usr/local/Ascend/ascend-toolkit/set_env.sh to ~/.bashrc
6、安装其他依赖项:
yum install blas-devel
7、安装pytorch依赖项:
pip3 install pyyaml
pip3 install typing_extensions
pip3 install wheel
pip3 install expecttest
pip3 install synr==0.5.0
pip3 install scipy tornado psutil attrs cloudpickle opencv-python -i https://pypi.tuna.tsinghua.edu.cn/simple
pip3 install sympy -i https://pypi.douban.com/simple/
8、安装官方torch包:
下载PyTorch v1.8.1源码包,1.11.0版本请替换版本号为v1.11.0,2.0.1版本请替换版本号为v2.0.1。
git clone -b v1.8.1 https://github.com/pytorch/pytorch.git --depth=1 pytorch_v1.8.1
进入源码包获取被动依赖代码。
cd pytorch_v1.8.1
git submodule sync
git submodule update --init --recursive
配置环境变量。
export USE_XNNPACK=0
执行编译安装。
#python3 setup.py install
python3 setup.py bdist_wheel
pip3 install pytorch_v1.8.1/dist/torch-1.8.0a0+56b43f4-cp38-cp38m-linux_aarch64.whl
# 下载对应PyTorch版本分支代码,进入插件根目录,以v1.8.1-5.0.rc2为例,其他版本请替换对应版本号
git clone -b v1.8.1-5.0.rc2 https://gitee.com/ascend/pytorch.git
cd pytorch
# 指定Python版本编包方式,以Python3.7为例,其他Python版本请使用 --python=3.8或--python3.9
bash ci/build.sh --python=3.8
安装pytorch/dist目录下生成的插件torch_npu包。
# 请用户根据实际情况更改命令中的torch_npu包名
pip3 install --upgrade dist/torch_npu-1.8.1.post2-cp38-cp38m-linux_aarch64.whl
pip3 install torchvision==0.9.1
9、测试安装成功:
python3 -c "import torch;import torch_npu; a = torch.randn(3, 4).npu(); print(a + a);"
10.MindX安装:
chmod +x Ascend-mindxsdk-mxvision_5.0.RC2_linux-aarch64.run
./Ascend-mindxsdk-mxvision_5.0.RC2_linux-aarch64.run --check
./Ascend-mindxsdk-mxvision_5.0.RC2_linux-aarch64.run --install
source MindX_SDK/mxVision/set_env.sh
references: