37 lines
1 KiB
Docker
37 lines
1 KiB
Docker
ARG TRTIS_IMAGE=nvcr.io/nvidia/tensorrtserver:20.02-py3
|
|
|
|
FROM ${TRTIS_IMAGE}
|
|
|
|
RUN mkdir -p /workspace/trt-tacotron2-waveglow
|
|
WORKDIR /workspace/trt-tacotron2-waveglow
|
|
|
|
# Download custom backend SDK
|
|
RUN wget https://github.com/NVIDIA/tensorrt-inference-server/releases/download/v1.11.0/v1.11.0_ubuntu1804.custombackend.tar.gz
|
|
RUN tar xf v1.11.0_ubuntu1804.custombackend.tar.gz && mv custom-backend-sdk ./trtis_sdk
|
|
|
|
# install cmake
|
|
RUN apt-get update && apt-get install -qy cmake && apt-get clean
|
|
|
|
# build the source code
|
|
ADD src/ "./src"
|
|
ADD CMakeLists.txt "./"
|
|
ADD configure "./"
|
|
|
|
RUN ./configure --trtis
|
|
RUN make
|
|
|
|
ARG TACOTRON2_MODEL="tacotron.json"
|
|
ARG WAVEGLOW_MODEL="waveglow.onnx"
|
|
ARG DENOISER_MODEL="denoiser.json"
|
|
|
|
RUN mkdir -p "/models" "/engines"
|
|
|
|
ADD "${TACOTRON2_MODEL}" /models/
|
|
ADD "${WAVEGLOW_MODEL}" /models/
|
|
ADD "${DENOISER_MODEL}" /models/
|
|
|
|
ADD model-config/tacotron2waveglow /models/tacotron2waveglow
|
|
RUN mkdir -p /models/tacotron2waveglow/1
|
|
RUN cp -v "./build/lib/libtt2i_trtis.so" /models/tacotron2waveglow/1/
|
|
|
|
ADD scripts "./scripts"
|