From c441eb08c44dfd4a7a8c68970fd3ebe7943d06ee Mon Sep 17 00:00:00 2001
From: shixian.shi <shixian.shi@alibaba-inc.com>
Date: 星期四, 09 三月 2023 15:26:03 +0800
Subject: [PATCH] tp_inference device bug
---
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