onnxruntime doesn't support gpu for mac

2 years ago · 24ae1c063e
parent ac2a9ec900
commit 24ae1c063e
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-gpu==1.10.0
+onnxruntime==1.10.0
 pandas
 paddlenlp
 paddlespeech_feat
--- a/setup.py
+++ b/setup.py
@ -44,7 +44,7 @@ base = [
    "loguru",
    "matplotlib",
    "nara_wpe",
-    "onnxruntime-gpu==1.10.0",
+    "onnxruntime==1.10.0",
    "pandas",
    "paddlenlp",
    "paddlespeech_feat",