From ac2a9ec9007ab54b479811ffbff0f7aa75c6b391 Mon Sep 17 00:00:00 2001
From: TianYuan <white-sky@qq.com>
Date: Wed, 3 Aug 2022 07:48:06 +0000
Subject: [PATCH] fix version of onnxruntime

---
 docs/requirements.txt | 2 +-
 setup.py              | 2 +-
 2 files changed, 2 insertions(+), 2 deletions(-)

diff --git a/docs/requirements.txt b/docs/requirements.txt
index bf1486c5..77dc609e 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 1cc82fa7..6809411f 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",
-- 
GitLab