Skip to content

Commit 784eb0d

Browse files
swethmandavaszmigacz
authored andcommitted
[BERT/TF] trtis dependency fix (NVIDIA#373)
1 parent 1b5a33e commit 784eb0d

File tree

3 files changed

+4
-9
lines changed

3 files changed

+4
-9
lines changed

.gitmodules

Lines changed: 0 additions & 4 deletions
Original file line numberDiff line numberDiff line change
@@ -1,10 +1,6 @@
11
[submodule "PyTorch/Translation/Transformer/cutlass"]
22
path = PyTorch/Translation/Transformer/cutlass
33
url = https://github.com/NVIDIA/cutlass.git
4-
[submodule "TensorFlow/LanguageModeling/BERT/tensorrt-inference-server"]
5-
path = TensorFlow/LanguageModeling/BERT/tensorrt-inference-server
6-
url = https://github.com/NVIDIA/tensorrt-inference-server.git
7-
branch = r19.08
84
[submodule "PyTorch/SpeechRecognition/Jasper/external/tensorrt-inference-server"]
95
path = PyTorch/SpeechRecognition/Jasper/external/tensorrt-inference-server
106
url = https://github.com/NVIDIA/tensorrt-inference-server.git

TensorFlow/LanguageModeling/BERT/Dockerfile

Lines changed: 4 additions & 4 deletions
Original file line numberDiff line numberDiff line change
@@ -1,10 +1,8 @@
11
ARG FROM_IMAGE_NAME=nvcr.io/nvidia/tensorflow:19.08-py3
22

3-
FROM tensorrtserver_client as trt
4-
53
FROM ${FROM_IMAGE_NAME}
64

7-
RUN apt-get update && apt-get install -y pbzip2 pv bzip2 libcurl3
5+
RUN apt-get update && apt-get install -y pbzip2 pv bzip2 libcurl4 curl
86

97
RUN pip install toposort networkx pytest nltk tqdm html2text progressbar
108

@@ -17,7 +15,9 @@ RUN git clone https://github.com/titipata/pubmed_parser
1715
RUN pip3 install /workspace/pubmed_parser
1816

1917
#Copy the perf_client over
20-
COPY --from=trt /workspace/install/ /workspace/install/
18+
ARG TRTIS_CLIENTS_URL=https://github.com/NVIDIA/tensorrt-inference-server/releases/download/v1.5.0/v1.5.0_ubuntu1804.clients.tar.gz
19+
RUN mkdir -p /workspace/install \
20+
&& curl -L ${TRTIS_CLIENTS_URL} | tar xvz -C /workspace/install
2121

2222
#Install the python wheel with pip
2323
RUN pip install /workspace/install/python/tensorrtserver*.whl
Lines changed: 0 additions & 1 deletion
This file was deleted.

0 commit comments

Comments
 (0)