ubuntu下配置OpenBLAS

最新推荐文章于 2025-09-11 01:45:57 发布

原创最新推荐文章于 2025-09-11 01:45:57 发布 · 2.8k 阅读

1 ·

CC 4.0 BY-SA版权

ubuntu 专栏收录该内容

1 篇文章

订阅专栏

本文介绍了如何在Ubuntu系统中配置OpenBLAS，包括下载编译、CMakeLists设置、应用注意事项等。特别指出，由于OpenBLAS与多线程应用可能存在冲突，应通过设置环境变量或编译选项限制为单线程运行。同时，当应用使用OpenMP时，编译时需要指定USE_OPENMP=1。最后，通过10000×10000矩阵相乘的测试，验证了OpenBLAS的性能。

OpenBLAS是高度优化的线性代数库。

1.下载编译

git clone https://github.com/xianyi/OpenBLAS.git
cd OpenBLAS
make
sudo make PREFIX=/usr/local/OpenBLAS install

2.CMakeLists

target_link_libraries(untitled1 /usr/local/OpenBLAS/lib/libopenblas.a)
链接blas动态库
target_link_libraries(untitled1 pthread)
线程操作需要pthread库

3.例子

void cblas_sgemm(CBLAS_LAYOUT layout, CBLAS_TRANSPOSE TransA,
                 CBLAS_TRANSPOSE TransB, const int M, const int N,
                 const int K, const float alpha, const float *A,
                 const int lda, const float *B, const int ldb,
                 const float beta, float *C, const int ldc);
用于计算C := alpha*op(A)*op(B) + beta*C。
如果alpha=1，beta=0，则计算2个矩阵相乘。
A是 M × K大小，B是K × N大小，C是M × N大小
由于openblas矩阵都是按照一维数组形式存储的，因此参数比较多。
layout：CblasRowMajor  1 2 3                 CblasColMajor   1 4 7
                       4 5 6                                 2 5 8
                       7 8 9                                 3 6 9
CblasRowMajor按行存储，CblasColMajor按列存储
TransA：CblasNoTrans 不进行变换，CblasTrans变为转置，CblasConjTrans变为共轭转置。
TransB：同上
M：A的行数，也就是C的行数，转置后的
N：B的列数，也就是C的列数，转置后的
K：A的列数，也就是B的行数，转置后的
alpha:一般为1
A：矩阵A
B：矩阵B
beta：一般为0
C：矩阵C
lda,ldb,ldc:如果是CblasColMajor则分别表示A，B，C的行，如果是CblasRowMajor，则分别表示A，B，C的列。

4.注意事项

如果应用是多线程的，将会与多线程下的OpenBLAS发生冲突。因此OpenBLAS只能运行在单线程下。可以通过三种方法设置：
1.在环境变量中OPENBLAS_NUM_THREADS=1
2.在应用运行时调用openblas_set_num_threads(1)
3.在编译时选择 make USE_THREAD=0
另外在应用被OpenMP优化时，编译时USE_OPENMP=1
5.测试
10000×10000的矩阵相乘，大概5s