Skip to content

Commit 7475648

Browse files
meatybobbynv-kkudrynski
authored andcommitted
[BERT/TF1] Fix Triton build issues for ARM
1 parent 260c8f6 commit 7475648

File tree

2 files changed

+3
-8
lines changed

2 files changed

+3
-8
lines changed

TensorFlow/LanguageModeling/BERT/Dockerfile

Lines changed: 2 additions & 7 deletions
Original file line numberDiff line numberDiff line change
@@ -15,13 +15,8 @@ RUN git clone https://github.com/titipata/pubmed_parser
1515

1616
RUN pip3 install /workspace/pubmed_parser
1717

18-
#Copy the perf_client over
19-
ARG TRTIS_CLIENTS_URL=https://github.com/NVIDIA/triton-inference-server/releases/download/v2.2.0/v2.2.0_ubuntu1804.clients.tar.gz
20-
RUN mkdir -p /workspace/install \
21-
&& curl -L ${TRTIS_CLIENTS_URL} | tar xvz -C /workspace/install
22-
23-
#Install the python wheel with pip
24-
RUN pip install /workspace/install/python/triton*.whl
18+
#Install tritonclient
19+
RUN pip install tritonclient[all]==2.5.0
2520

2621
WORKDIR /workspace/bert
2722
COPY . .

TensorFlow/LanguageModeling/BERT/triton/scripts/run_perf_client.sh

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -69,4 +69,4 @@ ARGS="\
6969

7070
echo "Using args: $(echo "$ARGS" | sed -e 's/ -/\n-/g')"
7171

72-
bash scripts/docker/launch.sh /workspace/install/bin/perf_client $ARGS
72+
bash scripts/docker/launch.sh perf_client $ARGS

0 commit comments

Comments
 (0)