test: add linux/arm64 to test matrix

Author: lsorber

.github/workflows/test.yml

@@ -11,11 +11,18 @@ jobs:
       matrix:
         python-version: ["3.8", "3.11"]
         cuda-version: ["11.8"]
+        platform: ["linux/amd64", "linux/arm64"]
 
     steps:
     - name: Checkout
       uses: actions/checkout@v3
 
+    - name: Set up QEMU
+      uses: docker/setup-qemu-action@v2
+
+    - name: Set up Docker Buildx
+      uses: docker/setup-buildx-action@v2
+
     - name: Login to Docker Hub
       uses: docker/login-action@v1 
       with:
@@ -29,6 +36,7 @@ jobs:
           "PYTHON_VERSION=${{ matrix.python-version }}"
           "CUDA_VERSION=${{ matrix.cuda-version }}"
         context: .
+        platforms: ${{ matrix.platform }}
         tags: radixai/python-gpu:${{ matrix.python-version }}-cuda${{ matrix.cuda-version }}
 
     - name: Test Docker image