fix version of onnxruntime

ac2a9ec9 · 小湉湉 · 070a08f2 · ac2a9ec9 · ac2a9ec9
隐藏空白更改
内联并排

Showing with 2 addition and 2 deletion

docs/requirements.txt docs/requirements.txt +1 -1

setup.py setup.py +1 -1

未找到文件。
--- 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",