diff --git a/examples/wenetspeech/asr1/local/test_wav.sh b/examples/wenetspeech/asr1/local/test_wav.sh index 13296af2e..90f92d80b 100755 --- a/examples/wenetspeech/asr1/local/test_wav.sh +++ b/examples/wenetspeech/asr1/local/test_wav.sh @@ -29,7 +29,7 @@ for type in attention_rescoring; do output_dir=${ckpt_prefix} mkdir -p ${output_dir} python3 -u ${BIN_DIR}/test_wav.py \ - --nproc ${ngpu} \ + --ngpu ${ngpu} \ --config ${config_path} \ --result_file ${output_dir}/${type}.rsl \ --checkpoint_path ${ckpt_prefix} \ diff --git a/tools/extras/install_kaldi.sh b/tools/extras/install_kaldi.sh index 3cdcd32d4..b93e7ecf6 100755 --- a/tools/extras/install_kaldi.sh +++ b/tools/extras/install_kaldi.sh @@ -9,6 +9,7 @@ apt-get install subversion -y KALDI_GIT="--depth 1 -b master https://github.com/kaldi-asr/kaldi.git" KALDI_DIR="$PWD/kaldi" +SHARED=false if [ ! -d "$KALDI_DIR" ]; then git clone $KALDI_GIT $KALDI_DIR @@ -23,17 +24,25 @@ git pull mkdir -p "python" touch "python/.use_default_python" +# check deps ./extras/check_dependencies.sh +# make tools make -j4 +# make src pushd ../src OPENBLAS_DIR=${KALDI_DIR}/../OpenBLAS mkdir -p ${OPENBLAS_DIR}/install -./configure --shared --use-cuda=no --static-math --mathlib=OPENBLAS --openblas-root=${OPENBLAS_DIR}/install +if [ $SHARED == true ]; + ./configure --shared --use-cuda=no --static-math --mathlib=OPENBLAS --openblas-root=${OPENBLAS_DIR}/install +else + ./configure --static --use-cuda=no --static-math --mathlib=OPENBLAS --openblas-root=${OPENBLAS_DIR}/install +fi make clean -j && make depend -j && make -j4 -popd +popd # kaldi/src + -popd +popd # kaldi/tools echo "Done installing Kaldi."