From 3d9f094e9652d4b84894c6fd4eae39a4a753b0f0 Mon Sep 17 00:00:00 2001
From: 游雁 <zhifu.gzf@alibaba-inc.com>
Date: 星期二, 16 五月 2023 23:48:00 +0800
Subject: [PATCH] train

---
 funasr/runtime/triton_gpu/Dockerfile/Dockerfile.server |    4 +++-
 1 files changed, 3 insertions(+), 1 deletions(-)

diff --git a/funasr/runtime/triton_gpu/Dockerfile/Dockerfile.server b/funasr/runtime/triton_gpu/Dockerfile/Dockerfile.server
index 459195c..d03610c 100644
--- a/funasr/runtime/triton_gpu/Dockerfile/Dockerfile.server
+++ b/funasr/runtime/triton_gpu/Dockerfile/Dockerfile.server
@@ -10,8 +10,10 @@
     cmake \
     libsndfile1
 
+# -i https://pypi.tuna.tsinghua.edu.cn/simple
+RUN pip3 install torch torchaudio 
 RUN pip3 install kaldifeat pyyaml
 
 # Dependency for client
-RUN pip3 install soundfile grpcio-tools tritonclient pyyaml
+RUN pip3 install soundfile grpcio-tools tritonclient
 WORKDIR /workspace

--
Gitblit v1.9.1