From 2b747626c898fb6b2ee78038ad052b761a77269d Mon Sep 17 00:00:00 2001
From: yijinsheng <1183186048@qq.com>
Date: 星期一, 11 十一月 2024 23:52:48 +0800
Subject: [PATCH] paraformer_large_offline triton运行bug 修复
---
runtime/triton_gpu/Dockerfile/Dockerfile.server | 15 +++++++++++----
1 files changed, 11 insertions(+), 4 deletions(-)
diff --git a/runtime/triton_gpu/Dockerfile/Dockerfile.server b/runtime/triton_gpu/Dockerfile/Dockerfile.server
index d03610c..b9f9c40 100644
--- a/runtime/triton_gpu/Dockerfile/Dockerfile.server
+++ b/runtime/triton_gpu/Dockerfile/Dockerfile.server
@@ -8,11 +8,18 @@
RUN apt-get update && apt-get -y install \
python3-dev \
cmake \
- libsndfile1
-
+ libsndfile1 \
+ language-pack-zh-hans
+ENV LANG="zh_CN.UTF-8"
+ENV LANGUAGE="zh_CN:zh:en_US:en"
# -i https://pypi.tuna.tsinghua.edu.cn/simple
-RUN pip3 install torch torchaudio
-RUN pip3 install kaldifeat pyyaml
+RUN pip3 config set global.index-url https://pypi.tuna.tsinghua.edu.cn/simple
+RUN pip3 install torch==2.4.1 torchaudio==2.4.1
+RUN pip3 install pyyaml
+
+##Refer to https://csukuangfj.github.io/kaldifeat/installation/from_wheels.html for installation.
+RUN pip3 install https://hf-mirror.com/csukuangfj/kaldifeat/resolve/main/ubuntu-cuda/kaldifeat-1.25.5.dev20240914+cuda12.1.torch2.4.1-cp38-cp38-manylinux_2_17_x86_64.manylinux2014_x86_64.whl
+
# Dependency for client
RUN pip3 install soundfile grpcio-tools tritonclient
--
Gitblit v1.9.1