游雁 3 سال پیش
والد
کامیت
77a7b40a7c
3فایلهای تغییر یافته به همراه45 افزوده شده و 1 حذف شده
  1. BIN
      docs/images/damo.png
  2. 6 1
      funasr/runtime/onnxruntime/readme.md
  3. 39 0
      funasr/runtime/onnxruntime/third_party/install_openblas.sh

BIN
docs/images/damo.png


+ 6 - 1
funasr/runtime/onnxruntime/readme.md

@@ -29,6 +29,7 @@ tester  /path/to/models/dir /path/to/wave/file
 
 ## 依赖
 - fftw3
+- openblas
 - onnxruntime
 
 ## 导出onnx格式模型文件
@@ -58,7 +59,11 @@ wget https://github.com/microsoft/onnxruntime/releases/download/v1.14.0/onnxrunt
 # onnxruntime-linux-x64-1.14.0  onnxruntime-linux-x64-1.14.0.tgz
 
 #install fftw3-dev
-apt install libfftw3-dev
+ubuntu: apt install libfftw3-dev
+centos: yum install fftw fftw-devel
+
+#install openblas
+bash ./third_party/install_openblas.sh
 
 # build
  cmake  -DCMAKE_BUILD_TYPE=release .. -DONNXRUNTIME_DIR=/mnt/c/Users/ma139/RapidASR/cpp_onnx/build/onnxruntime-linux-x64-1.14.0

+ 39 - 0
funasr/runtime/onnxruntime/third_party/install_openblas.sh

@@ -0,0 +1,39 @@
+#!/usr/bin/env bash
+
+OPENBLAS_VERSION=0.3.13
+
+WGET=${WGET:-wget}
+
+set -e
+
+if ! command -v gfortran 2>/dev/null; then
+  echo "$0: gfortran is not installed.  Please install it, e.g. by:"
+  echo " apt-get install gfortran"
+  echo "(if on Debian or Ubuntu), or:"
+  echo " yum install gcc-gfortran"
+  echo "(if on RedHat/CentOS).  On a Mac, if brew is installed, it's:"
+  echo " brew install gfortran"
+  exit 1
+fi
+
+
+tarball=OpenBLAS-$OPENBLAS_VERSION.tar.gz
+
+rm -rf xianyi-OpenBLAS-* OpenBLAS OpenBLAS-*.tar.gz
+
+if [ -d "$DOWNLOAD_DIR" ]; then
+  cp -p "$DOWNLOAD_DIR/$tarball" .
+else
+  url=$($WGET -qO- "https://api.github.com/repos/xianyi/OpenBLAS/releases/tags/v${OPENBLAS_VERSION}" | python -c 'import sys,json;print(json.load(sys.stdin)["tarball_url"])')
+  test -n "$url"
+  $WGET -t3 -nv -O $tarball "$url"
+fi
+
+tar xzf $tarball
+mv xianyi-OpenBLAS-* OpenBLAS
+
+make PREFIX=$(pwd)/OpenBLAS/install USE_LOCKING=1 USE_THREAD=0 -C OpenBLAS all install
+if [ $? -eq 0 ]; then
+   echo "OpenBLAS is installed successfully."
+   rm $tarball
+fi