diff --git a/docs/requirements.txt b/docs/requirements.txt
index bf1486c5e22fb3e08b5dd069810b54755f10505e..77dc609e7253bbcaf65c27cda019cb195f8a3a0c 100644
--- 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
diff --git a/setup.py b/setup.py
index 1cc82fa76f0ee4e7a404675b5dd35b84884b50da..6809411f71b4d394685f4bac4422443a0e96f769 100644
--- 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",