maxmustermann/DeepLearningExamples
1
0
Fork 0
Code Issues Pull requests Projects Releases Wiki Activity

[BERT/TF] trtis dependency fix (#373)

Browse source
This commit is contained in:
Swetha Mandava 2020-01-13 13:37:36 -08:00 committed by Szymon Migacz
parent 1b5a33ee00
commit 784eb0d8ca
3 changed files with 4 additions and 9 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

4
.gitmodules vendored
View file

@ -1,10 +1,6 @@
[submodule "PyTorch/Translation/Transformer/cutlass"]
path = PyTorch/Translation/Transformer/cutlass
url = https://github.com/NVIDIA/cutlass.git
[submodule "TensorFlow/LanguageModeling/BERT/tensorrt-inference-server"]
path = TensorFlow/LanguageModeling/BERT/tensorrt-inference-server
url = https://github.com/NVIDIA/tensorrt-inference-server.git
branch = r19.08
[submodule "PyTorch/SpeechRecognition/Jasper/external/tensorrt-inference-server"]
path = PyTorch/SpeechRecognition/Jasper/external/tensorrt-inference-server
url = https://github.com/NVIDIA/tensorrt-inference-server.git

8
TensorFlow/LanguageModeling/BERT/Dockerfile
View file

@ -1,10 +1,8 @@
ARG FROM_IMAGE_NAME=nvcr.io/nvidia/tensorflow:19.08-py3
FROM tensorrtserver_client as trt
FROM ${FROM_IMAGE_NAME}
RUN apt-get update && apt-get install -y pbzip2 pv bzip2 libcurl3
RUN apt-get update && apt-get install -y pbzip2 pv bzip2 libcurl4 curl
RUN pip install toposort networkx pytest nltk tqdm html2text progressbar
@ -17,7 +15,9 @@ RUN git clone https://github.com/titipata/pubmed_parser
RUN pip3 install /workspace/pubmed_parser
#Copy the perf_client over
COPY --from=trt /workspace/install/ /workspace/install/
ARG TRTIS_CLIENTS_URL=https://github.com/NVIDIA/tensorrt-inference-server/releases/download/v1.5.0/v1.5.0_ubuntu1804.clients.tar.gz
RUN mkdir -p /workspace/install \
&& curl -L ${TRTIS_CLIENTS_URL} | tar xvz -C /workspace/install
#Install the python wheel with pip
RUN pip install /workspace/install/python/tensorrtserver*.whl

1
TensorFlow/LanguageModeling/BERT/tensorrt-inference-server

@ -1 +0,0 @@
Subproject commit 33931a4f884ea91e92c0c292473d8cbd47f3cafd