Update for ARM support (#229)

Author: matthewkotila

Dockerfile

@@ -1,4 +1,4 @@
-# Copyright (c) 2020, NVIDIA CORPORATION. All rights reserved.
+# Copyright (c) 2020-2021, NVIDIA CORPORATION & AFFILIATES. All rights reserved.
 #
 # Licensed under the Apache License, Version 2.0 (the "License");
 # you may not use this file except in compliance with the License.
@@ -36,17 +36,20 @@ RUN apt-get update && \
 RUN mkdir -p /opt/triton-model-analyzer
 
 # Install DCGM
-RUN apt-get update && apt-get install -y --no-install-recommends software-properties-common && \
-      wget https://developer.download.nvidia.com/compute/cuda/repos/ubuntu2004/x86_64/cuda-ubuntu2004.pin && \
+ARG TARGETARCH
+RUN if [ "${TARGETARCH}" = "amd64" ] ; then ARCH_DIR="x86_64" ; fi ; \
+    if [ "${TARGETARCH}" = "arm64" ] ; then ARCH_DIR="sbsa" ; fi ; \
+    apt-get update && apt-get install -y --no-install-recommends software-properties-common && \
+      wget https://developer.download.nvidia.com/compute/cuda/repos/ubuntu2004/${ARCH_DIR}/cuda-ubuntu2004.pin && \
       mv cuda-ubuntu2004.pin /etc/apt/preferences.d/cuda-repository-pin-600 && \
-      apt-key adv --fetch-keys https://developer.download.nvidia.com/compute/cuda/repos/ubuntu2004/x86_64/7fa2af80.pub && \
-      add-apt-repository "deb https://developer.download.nvidia.com/compute/cuda/repos/ubuntu2004/x86_64/ /" && \
+      apt-key adv --fetch-keys https://developer.download.nvidia.com/compute/cuda/repos/ubuntu2004/${ARCH_DIR}/7fa2af80.pub && \
+      add-apt-repository "deb https://developer.download.nvidia.com/compute/cuda/repos/ubuntu2004/${ARCH_DIR}/ /" && \
       apt-get install -y datacenter-gpu-manager=1:${DCGM_VERSION}
 
 # Install tritonclient
 COPY --from=sdk /workspace/install/python /tmp/tritonclient
 RUN find /tmp/tritonclient -maxdepth 1 -type f -name \
-    "tritonclient-*-manylinux1_x86_64.whl" | xargs printf -- '%s[all]' | \
+    "tritonclient-*-manylinux*.whl" | xargs printf -- '%s[all]' | \
     xargs pip3 install --upgrade && rm -rf /tmp/tritonclient/
 
 WORKDIR /opt/triton-model-analyzer
@@ -61,7 +64,7 @@ RUN chmod +x build_wheel.sh && \
     rm -f perf_analyzer
 RUN python3 -m pip install --upgrade pip && \
     python3 -m pip install nvidia-pyindex && \
-    python3 -m pip install wheels/triton_model_analyzer-*-manylinux1_x86_64.whl
+    python3 -m pip install wheels/triton_model_analyzer-*-manylinux*.whl
 
 RUN apt-get install -y wkhtmltopdf