ソースを参照

[BERT/TF1] Fix Triton build issues for ARM

Bobby Chen 3 年 前
コミット
747564894c

+ 2 - 7
TensorFlow/LanguageModeling/BERT/Dockerfile

@@ -15,13 +15,8 @@ RUN git clone https://github.com/titipata/pubmed_parser
 
 RUN pip3 install /workspace/pubmed_parser
 
-#Copy the perf_client over
-ARG TRTIS_CLIENTS_URL=https://github.com/NVIDIA/triton-inference-server/releases/download/v2.2.0/v2.2.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/triton*.whl
+#Install tritonclient
+RUN pip install tritonclient[all]==2.5.0
 
 WORKDIR /workspace/bert
 COPY . .

+ 1 - 1
TensorFlow/LanguageModeling/BERT/triton/scripts/run_perf_client.sh

@@ -69,4 +69,4 @@ ARGS="\
 
 echo "Using args:  $(echo "$ARGS" | sed -e 's/   -/\n-/g')"
 
-bash scripts/docker/launch.sh /workspace/install/bin/perf_client $ARGS
+bash scripts/docker/launch.sh perf_client $ARGS