拓展安装(开启BLAS加速)
BLAS加速用来加速oneDNN中涉及的线性代数运算,若您需要此加速功能请参照本节内容进行安装。安装过程中请根据实际情况替换安装路径。
- 获取ArmPL安装包。
若服务器可以访问网络,则可以直接在服务器上使用wget命令下载源码。
- 进入“/path/to/ArmPL”目录。
1
cd /path/to/ArmPL
- 下载源码。
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
- 进入“/path/to/ArmPL”目录。
- 进入“/path/to/ArmPL”目录。
1
cd /path/to/ArmPL
- 创建安装目录。
1
mkdir install
- 解压ArmPL安装包。
1
tar -xvf arm-performance-libraries_23.10_RHEL-8_gcc-10.4.tar
- 进入“arm-performance-libraries_23.10_RHEL-8”目录。
1
cd arm-performance-libraries_23.10_RHEL-8
- 安装。
1
./arm-performance-libraries_23.10_RHEL-8.sh -i /path/to/ArmPL/install -a
安装参数说明:
- -i:指定ArmPL库的安装路径。
- -a:自动接受EULA(End User License Agreement)最终用户许可协议。
- 设置环境变量。
1
export MODULEPATH=$MODULEPATH:/path/to/ArmPL/install/modulefiles
- 加载module。
1
module load armpl/23.10.0_gcc-10.4
- 安装oneDNN,请参见基本安装的安装流程,将步骤7替换为以下命令,完成oneDNN编译安装。
1
cmake -DCMAKE_INSTALL_PREFIX=/path/to/oneDNN-3.3.3/build/install/ -DONEDNN_BLAS_VENDOR=ARMPL ..
- 安装完毕后,可通过以下命令查看libdnnl.so是否链接到了ArmPL库的so。
1
ldd /path/to/oneDNN-3.3.3/build/install/lib64/libdnnl.so
回显如下图所示表示libdnnl.so已经链接到了ArmPL库的so。
父主题: 源码编译安装