Require image for ARM64 architecture

Signed-off-by: odidev <odidev@puresoftware.com>

.github/workflows/go.yml

@@ -2,9 +2,11 @@ name: Go
 
 on:
   push:
-    branches: [ main ]
+    branches: [ local_testing ]
   pull_request:
-    branches: [ main ]
+    branches: [ local_testing ]
+    tags:
+      - v*
 
 jobs:
 
@@ -23,3 +25,27 @@ jobs:
 
     - name: Lint
       run: make check-format
+
+    - uses: actions/checkout@v2
+    - name: Prepare
+      id: prepare
+      run: |
+          TAG=${GITHUB_REF##*/}
+          echo ::set-output name=tag_name::${TAG}
+    - name: Set up QEMU
+      uses: docker/setup-qemu-action@v1
+    - name: Set up Docker Buildx
+      uses: docker/setup-buildx-action@v1
+    - name: Login do docker.io
+      run: docker login -u odidev -p ${{ secrets.DOCKER_TOKEN }}
+    - name: build and publish image
+      id: docker_build
+      uses: docker/build-push-action@v2
+      with:
+        context: .
+        file: ./Dockerfile
+        platforms: linux/amd64,linux/arm64
+        push: true
+        tags: |
+          odidev/dcgm-exporter:${{ steps.prepare.outputs.tag_name }}
+          odidev/dcgm-exporter:latest

Dockerfile

@@ -5,13 +5,19 @@ ENV DEBIAN_FRONTEND=noninteractive
 
 RUN apt-get update && apt-get install -y --no-install-recommends \
     gnupg2 curl ca-certificates build-essential && \
+    if [ `uname -m` = "aarch64" ] ; then \
+    curl -fsSL https://developer.download.nvidia.com/compute/cuda/repos/ubuntu2004/sbsa/7fa2af80.pub | apt-key add - && \
+    curl -s https://storage.googleapis.com/golang/go1.16.5.linux-arm64.tar.gz| tar -v -C /usr/local -xz && \
+    echo "deb https://developer.download.nvidia.com/compute/cuda/repos/ubuntu2004/sbsa /" > /etc/apt/sources.list.d/cuda.list && \
+    echo "deb https://developer.download.nvidia.com/compute/machine-learning/repos/ubuntu2004/sbsa /" > /etc/apt/sources.list.d/nvidia-ml.list; \
+    else \
     curl -fsSL https://developer.download.nvidia.com/compute/cuda/repos/ubuntu2004/x86_64/7fa2af80.pub | apt-key add - && \
     curl -s https://storage.googleapis.com/golang/go1.16.5.linux-amd64.tar.gz| tar -v -C /usr/local -xz && \
     echo "deb https://developer.download.nvidia.com/compute/cuda/repos/ubuntu2004/x86_64 /" > /etc/apt/sources.list.d/cuda.list && \
-    echo "deb https://developer.download.nvidia.com/compute/machine-learning/repos/ubuntu2004/x86_64 /" > /etc/apt/sources.list.d/nvidia-ml.list && \
+    echo "deb https://developer.download.nvidia.com/compute/machine-learning/repos/ubuntu2004/x86_64 /" > /etc/apt/sources.list.d/nvidia-ml.list; \
+    fi && \
     apt-get purge --autoremove -y curl \
     && rm -rf /var/lib/apt/lists/*
-
 RUN apt-get update && apt-get install -y --no-install-recommends \
     datacenter-gpu-manager