Ascend昇腾驱动和固件安装

昇腾驱动和固件安装

1. 安装依赖

   sudo yum install -y gcc gcc-c++ make cmake unzip zlib-devel libffi-devel openssl-devel pciutils net-tools sqlite-devel lapack-devel gcc-gfortran python3-devel
   

2. 创建驱动运行用户

   sudo groupadd -g HwHiAiUser
   sudo useradd -g HwHiAiUser -d /home/HwHiAiUser -m HwHiAiUser -s /bin/bash
   sudo usermod -aG HwHiAiUser $USER
   

3. 下载安装驱动

   wget "https://ascend-repo.obs.cn-east-2.myhuaweicloud.com/Ascend HDK/Ascend HDK 23.0.3/Ascend-hdk-910b-npu-driver_23.0.3_linux-aarch64.run"
   sudo sh Ascend-hdk-910b-npu-driver_23.0.3_linux-aarch64.run --full --install-for-all
   

4. 验证

   npu-smi info
   

5. 安装固件

   wget "https://ascend-repo.obs.cn-east-2.myhuaweicloud.com/Ascend HDK/Ascend HDK 23.0.3/Ascend-hdk-910b-npu-firmware_7.1.0.5.220.run"
   sudo sh Ascend-hdk-910b-npu-firmware_7.1.0.5.220.run --full
   

6. 安装python依赖

   pip3 install -i https://pypi.tuna.tsinghua.edu.cn/simple attrs numpy decorator sympy cffi pyyaml pathlib2 psutil protobuf scipy requests absl-py wheel typing_extensions
   

7. NPU开发包

   # replace the url according to your CANN version and devices
   # install CANN Toolkit
   wget https://ascend-repo.obs.cn-east-2.myhuaweicloud.com/Milan-ASL/Milan-ASL%20V100R001C17SPC701/Ascend-cann-toolkit_8.0.RC1.alpha001_linux-"$(uname -i)".run
   bash Ascend-cann-toolkit_8.0.RC1.alpha001_linux-"$(uname -i)".run --install
   
   # install CANN Kernels
   wget https://ascend-repo.obs.cn-east-2.myhuaweicloud.com/Milan-ASL/Milan-ASL%20V100R001C17SPC701/Ascend-cann-kernels-910b_8.0.RC1.alpha001_linux.run
   bash Ascend-cann-kernels-910b_8.0.RC1.alpha001_linux.run --install
   
   # set env variables
   source /usr/local/Ascend/ascend-toolkit/set_env.sh
   
   ===========
   = Summary =
   ===========
   
   Driver:   Installed in /usr/local/Ascend/driver.
   Toolkit:  Ascend-cann-toolkit_8.0.RC1.alpha001_linux-aarch64 install success, installed in /usr/local/Ascend.
   
   Please make sure that the environment variables have been configured.
   -  To take effect for all users, you can add "source /usr/local/Ascend/ascend-toolkit/set_env.sh" to /etc/profile.
   -  To take effect for current user, you can exec command below: source /usr/local/Ascend/ascend-toolkit/set_env.sh or add "source /usr/local/Ascend/ascend-toolkit/set_env.sh" to ~/.bashrc.
   

![在这里插入图片描述](https://i-blog.csdnimg.cn/direct/2c2506e908d94751b7464f1fa1ce16e6.png)

8. torch-npu安装

 ```bash
 git clone --depth 1 https://github.com/hiyouga/LLaMA-Factory.git
 cd LLaMA-Factory
 pip install -e ".[torch-npu,metrics]"
 ```

 - 校验

   ```bash
   import torch
   import torch_npu
   print(torch.npu.is_available())
   ```

9. 卸载

   • 卸载驱动

     sudo /usr/local/Ascend/driver/script/uninstall.sh
     

   • 卸载固件

     sudo /usr/local/Ascend/firmware/script/uninstall.sh
     

   • 卸载CANN-toolkit

     ~/Ascend/ascend-toolkit/<cann_version>/{arch}-linux/script/uninstall.sh