鲲鹏社区首页
中文
注册
我要评分
文档获取效率
文档正确性
内容完整性
文档易理解
在线提单
论坛求助

拓展安装(开启BLAS加速)

BLAS加速用来加速oneDNN中涉及的线性代数运算,若您需要此加速功能请参照本节内容进行安装。安装过程中请根据实际情况替换安装路径。

  1. 获取ArmPL安装包。
    1. 下载ArmPL安装包arm-performance-libraries_23.10_RHEL-8_gcc-10.4.tar
    2. 使用SFTP工具将源码上传至服务器“/path/to/ArmPL”目录。

    若服务器可以访问网络,则可以直接在服务器上使用wget命令下载源码。

    1. 进入“/path/to/ArmPL”目录。
      1
      cd /path/to/ArmPL
      
    2. 下载源码。
      1
      wget https://developer.arm.com/-/media/Files/downloads/hpc/arm-performance-libraries/23-10/rhel-8/arm-performance-libraries_23.10_RHEL-8_gcc-10.4.tar --no-check-certificate
      
  2. 进入“/path/to/ArmPL”目录。
    1
    cd /path/to/ArmPL
    
  3. 创建安装目录。
    1
    mkdir install
    
  4. 解压ArmPL安装包。
    1
    tar -xvf arm-performance-libraries_23.10_RHEL-8_gcc-10.4.tar
    
  5. 进入“arm-performance-libraries_23.10_RHEL-8”目录。
    1
    cd arm-performance-libraries_23.10_RHEL-8
    
  6. 安装。
    1
    ./arm-performance-libraries_23.10_RHEL-8.sh -i /path/to/ArmPL/install -a
    

    安装参数说明:

    • -i:指定ArmPL库的安装路径。
    • -a:自动接受EULA(End User License Agreement)最终用户许可协议。
  7. 设置环境变量。
    1
    export MODULEPATH=$MODULEPATH:/path/to/ArmPL/install/modulefiles
    
  8. 加载module。
    1
    module load armpl/23.10.0_gcc-10.4
    
  9. 安装oneDNN,请参见基本安装的安装流程,将步骤7替换为以下命令,完成oneDNN编译安装。
    1
    cmake -DCMAKE_INSTALL_PREFIX=/path/to/oneDNN-3.3.3/build/install/ -DONEDNN_BLAS_VENDOR=ARMPL ..
    
  10. 安装完毕后,可通过以下命令查看libdnnl.so是否链接到了ArmPL库的so。
    1
    ldd /path/to/oneDNN-3.3.3/build/install/lib64/libdnnl.so
    

    回显如下图所示表示libdnnl.so已经链接到了ArmPL库的so。