fix version of onnxruntime

2 years ago · ac2a9ec900
parent 070a08f2be
commit ac2a9ec900
2 changed files with 2 additions and 2 deletions
--- a/docs/requirements.txt
+++ b/docs/requirements.txt
@ -18,7 +18,7 @@ librosa==0.8.1
 loguru
 matplotlib
 nara_wpe
-onnxruntime
+onnxruntime-gpu==1.10.0
 pandas
 paddlenlp
 paddlespeech_feat
--- a/setup.py
+++ b/setup.py
@ -44,7 +44,7 @@ base = [
    "loguru",
    "matplotlib",
    "nara_wpe",
-    "onnxruntime",
+    "onnxruntime-gpu==1.10.0",
    "pandas",
    "paddlenlp",
    "paddlespeech_feat",