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

开启BLAS加速

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

  1. 使用PuTTY工具,以root用户登录服务器。
  2. 执行以下命令安装OpenBLAS。
    1
    yum install openblas gcc-gfortran
    
  3. (可选)如果安装过PyTorch,需要先清理旧的编译文件。
    1. 进入“/path/to/pytorch”目录。
      1
      cd /path/to/pytorch
      
    2. 删除构建目录。
      1
      rm -rf ./build
      
  4. 请参见基本安装完成PyTorch的安装。

    BLAS库不需要设置环境变量,编译PyTorch时会自动检测用户环境上的BLAS库。以OpenBLAS为例,如果下载了OpenBLAS,构建后可以看到PyTorch启用了OpenBLAS。

    如果设置了BLAS的环境变量,则意味着强制开启BLAS库,如果没有搜索到可用的BLAS库,将会导致编译失败。

  5. 使用如下命令检查是否已开启BLAS特性。
    1
    ldd build/lib/libtorch.so | grep blas
    

    回显如图所示表示已经开启了BLAS特性。