开启BLAS加速
BLAS加速是用来加速线性代数运算,若您需要此加速功能请参照本节内容进行安装。安装过程中请根据实际情况替换安装路径。
- 使用PuTTY工具,以root用户登录服务器。
- 执行以下命令安装OpenBLAS。
1
yum install openblas gcc-gfortran
- (可选)如果安装过PyTorch,需要先清理旧的编译文件。
- 进入“/path/to/pytorch”目录。
1
cd /path/to/pytorch
- 删除构建目录。
1
rm -rf ./build
- 进入“/path/to/pytorch”目录。
- 请参见基本安装完成PyTorch的安装。
BLAS库不需要设置环境变量,编译PyTorch时会自动检测用户环境上的BLAS库。以OpenBLAS为例,如果下载了OpenBLAS,构建后可以看到PyTorch启用了OpenBLAS。
如果设置了BLAS的环境变量,则意味着强制开启BLAS库,如果没有搜索到可用的BLAS库,将会导致编译失败。
- 使用如下命令检查是否已开启BLAS特性。
1
ldd build/lib/libtorch.so | grep blas
回显如图所示表示已经开启了BLAS特性。
父主题: 源码编译安装