From 2868fe3df4e92a6ae3e327faf6e57ea492e04124 Mon Sep 17 00:00:00 2001 From: 志浩 <neo.dzh@alibaba-inc.com> Date: 星期四, 16 三月 2023 19:24:21 +0800 Subject: [PATCH] Merge branch 'main' into dev_dzh --- funasr/runtime/triton_gpu/Dockerfile/Dockerfile.server | 19 +++++++++++++++++++ 1 files changed, 19 insertions(+), 0 deletions(-) diff --git a/funasr/runtime/triton_gpu/Dockerfile/Dockerfile.server b/funasr/runtime/triton_gpu/Dockerfile/Dockerfile.server new file mode 100644 index 0000000..d03610c --- /dev/null +++ b/funasr/runtime/triton_gpu/Dockerfile/Dockerfile.server @@ -0,0 +1,19 @@ +FROM nvcr.io/nvidia/tritonserver:23.01-py3 +# https://docs.nvidia.com/deeplearning/frameworks/support-matrix/index.html +# Please choose previous tritonserver:xx.xx if you encounter cuda driver mismatch issue + +LABEL maintainer="NVIDIA" +LABEL repository="tritonserver" + +RUN apt-get update && apt-get -y install \ + python3-dev \ + 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 +WORKDIR /workspace -- Gitblit v1.9.1