Update Dockerfile-arm64 (ultralytics#7860)

* Update Dockerfile-arm64 * Update docker.yml
ctjanuhowski · Sep 8, 2022 · 7f66fdf · 7f66fdf
1 parent 2dd661a
commit 7f66fdf
Show file tree

Hide file tree

Showing 2 changed files with 11 additions and 10 deletions.
diff --git a/.github/workflows/docker.yml b/.github/workflows/docker.yml
@@ -6,7 +6,7 @@ on:
   push:
     branches: [master]
   # pull_request:
-  #  branches: [master]
+  #   branches: [master]
 
 jobs:
   docker:
@@ -44,11 +44,11 @@ jobs:
           push: true
           tags: ultralytics/yolov5:latest
 
-      #- name: Build and push arm64 image
-      #  uses: docker/build-push-action@v3
-      #  with:
-      #    context: .
-      #    platforms: linux/arm64
-      #    file: utils/docker/Dockerfile-arm64
-      #    push: true
-      #    tags: ultralytics/yolov5:latest-arm64
+      - name: Build and push arm64 image
+        uses: docker/build-push-action@v3
+        with:
+          context: .
+          platforms: linux/arm64
+          file: utils/docker/Dockerfile-arm64
+          push: true
+          tags: ultralytics/yolov5:latest-arm64
diff --git a/utils/docker/Dockerfile-arm64 b/utils/docker/Dockerfile-arm64
@@ -10,7 +10,8 @@ ADD https://ultralytics.com/assets/Arial.ttf https://ultralytics.com/assets/Aria
 # Install linux packages
 RUN apt update
 RUN DEBIAN_FRONTEND=noninteractive TZ=Etc/UTC apt install -y tzdata
-RUN apt install --no-install-recommends -y python3-pip git zip curl htop libgl1-mesa-glx libglib2.0-0 libpython3.8-dev
+RUN apt install --no-install-recommends -y python3-pip git zip curl htop gcc \
+    libgl1-mesa-glx libglib2.0-0 libpython3.8-dev
 # RUN alias python=python3
 
 # Install pip packages