华为atlas300安装教程

发布时间:2023年12月20日

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);"

references:

昇腾社区-官网丨昇腾万里 让智能无所不及

文章来源:https://blog.csdn.net/qq_14845119/article/details/135104332
本文来自互联网用户投稿,该文观点仅代表作者本人,不代表本站立场。本站仅提供信息存储空间服务,不拥有所有权,不承担相关法律责任。