首页
/ 在Linux系统上构建与安装hipBLASLt的完整指南

在Linux系统上构建与安装hipBLASLt的完整指南

2025-06-09 01:04:51作者:滕妙奇

前言

hipBLASLt是ROCm生态系统中针对高性能计算优化的BLAS库精简版本,专为AMD GPU设计。本文将详细介绍在Linux系统上安装hipBLASLt的多种方法,包括使用预构建包和从源代码构建两种主要方式。

系统要求

在开始安装前,请确保您的系统满足以下要求:

  1. 支持ROCm的平台环境
  2. 兼容版本的hipBLAS库
  3. 基本的开发工具链(如gcc、make等)
  4. CMake构建工具(特定版本要求见下文)

方法一:安装预构建包

对于大多数用户而言,使用系统包管理器安装预构建的hipBLASLt是最简单快捷的方式。

在基于Debian的系统(如Ubuntu)上,执行:

sudo apt update
sudo apt install hipblaslt

这种安装方式会自动处理所有依赖关系,适合快速部署场景。

方法二:使用install.sh脚本构建

hipBLASLt项目提供了一个便捷的install.sh脚本,可以自动化构建和安装过程。

基本用法

  1. 获取帮助信息:

    ./install.sh -h
    
  2. 构建库及其依赖:

    ./install.sh -d
    

    (注:-d选项只需在首次构建时使用)

  3. 仅构建库(假设依赖已安装):

    ./install.sh
    
  4. 构建并全局安装:

    ./install.sh -i
    

构建客户端工具

hipBLASLt包含两个有用的客户端工具:

  • hipblaslt-test:用于运行GoogleTest测试
  • hipblaslt-bench:用于基准测试单个函数

构建客户端工具的常用命令:

  1. 构建所有组件(库+客户端+依赖):

    ./install.sh -dc
    
  2. 仅构建客户端(假设依赖已安装):

    ./install.sh -c
    
  3. 构建并全局安装所有组件:

    ./install.sh -idc
    

静态库构建

要构建静态库版本,可使用--static选项:

./install.sh --static

注意:静态库构建会产生额外的运行时依赖,可能需要设置HIPBLASLT_TENSILE_LIBPATH环境变量。

方法三:手动构建

对于需要更多控制的高级用户,可以手动配置和构建hipBLASLt。

基础库构建步骤

  1. 创建构建目录:

    mkdir -p build/release
    cd build/release
    
  2. 配置CMake(使用AMD编译器):

    CXX=/opt/rocm/bin/amdclang++ ccmake [HIPBLASLT_SOURCE]
    
  3. 构建和安装:

    make -j$(nproc)
    sudo make install
    

客户端工具构建

客户端工具需要额外依赖:

  1. LAPACK(需要Fortran编译器)
  2. GoogleTest框架

构建步骤:

  1. 可选:构建依赖项

    mkdir -p build/release/deps
    cd build/release/deps
    ccmake -DBUILD_BOOST=OFF [HIPBLASLT_SOURCE]/deps
    make -j$(nproc) install
    
  2. 配置客户端构建:

    CXX=/opt/rocm/bin/amdclang++ ccmake -DBUILD_CLIENTS_TESTS=ON -DBUILD_CLIENTS_BENCHMARKS=ON [HIPBLASLT_SOURCE]
    
  3. 构建和安装:

    make -j$(nproc)
    sudo make install
    

常见问题与建议

  1. 权限问题:全局安装需要sudo权限,如无必要可考虑本地安装
  2. 依赖管理:建议使用系统包管理器优先安装已知依赖
  3. 构建类型:生产环境建议使用Release构建以获得最佳性能
  4. 编译器选择:推荐使用AMD优化的编译器(amdclang++)

结语

本文介绍了hipBLASLt的多种安装方式,用户可根据自身需求选择最适合的方法。对于大多数应用场景,预构建包或install.sh脚本已能满足需求;而对于需要深度定制或开发的环境,手动构建提供了最大的灵活性。

登录后查看全文
热门项目推荐