File tree Expand file tree Collapse file tree 2 files changed +3
-8
lines changed
TensorFlow/LanguageModeling/BERT Expand file tree Collapse file tree 2 files changed +3
-8
lines changed Original file line number Diff line number Diff line change @@ -15,13 +15,8 @@ RUN git clone https://github.com/titipata/pubmed_parser
1515
1616RUN 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
2621WORKDIR /workspace/bert
2722COPY . .
Original file line number Diff line number Diff line change @@ -69,4 +69,4 @@ ARGS="\
6969
7070echo " 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
You can’t perform that action at this time.
0 commit comments