进行Ascend C算子开发之前,需要安装驱动固件和CANN软件包,完成环境准备。
驱动固件(仅昇腾设备需要)的安装步骤可参见《CANN 软件安装指南》的“安装NPU驱动固件”章节,本节不再给出安装示例。
本节以训练&推理&开发调试业务场景、操作系统架构X86_64、root用户操作为例,介绍CANN软件包安装的步骤,其他场景的安装步骤类似,详细的可参见《CANN 软件安装指南》。
1 | apt-get install -y gcc make net-tools cmake python3 python3-dev python3-pip |
1 | yum install -y gcc make net-tools cmake python3 python3-devel python3-pip |
运行以下命令检查版本:
1 2 3 4 | python3 --version pip3 --version cmake --version gcc --version |
如果版本不符合要求:
1 | pip3 install attrs cython numpy==1.24.0 decorator sympy cffi pyyaml pathlib2 psutil protobuf==3.20 scipy requests absl-py |
当前命令会安装最新或指定版本的依赖,如果安装过程中报错,或需要了解更详细的版本要求,请参考《CANN 软件安装指南》的“依赖列表”。
chmod +x Ascend-cann-toolkit_XXX_linux-x86_64.run
./Ascend-cann-toolkit_XXX_linux-x86_64.run --check
./Ascend-cann-toolkit_XXX_linux-x86_64.run --install
用户需签署华为企业业务最终用户许可协议(EULA)后进入安装流程,根据回显页面输入y或Y接受协议,安装完成后,若显示如下信息,则说明软件安装成功。
1 | [INFO] Ascend-cann-toolkit install success |
1 | source /usr/local/Ascend/ascend-toolkit/set_env.sh |
mkdir -p cmake-3.16 && wget -qO- "https://cmake.org/files/v3.16/cmake-3.16.0-linux-x86_64.tar.gz" | tar --strip-components=1 -xz -C cmake-3.16 export PATH=`pwd`/cmake-3.16/bin:$PATH