project-lakitu: cos-extensions: support GPU driver installation on ARM64

BUG=b/349468635, b/345487871

Change-Id: Icd36fd5f8e924c3d14ca38ee70eb6e0f933d2b53
Reviewed-on: https://cos-review.googlesource.com/c/cos/overlays/board-overlays/+/77864
Reviewed-by: Oleksandr Tymoshenko <ovt@google.com>
Tested-by: Cusky Presubmit Bot <presubmit@cos-infra-prod.iam.gserviceaccount.com>
diff --git a/project-lakitu/app-admin/extensions-manager/extensions-manager-0.0.1-r49.ebuild b/project-lakitu/app-admin/extensions-manager/extensions-manager-0.0.1-r50.ebuild
similarity index 100%
rename from project-lakitu/app-admin/extensions-manager/extensions-manager-0.0.1-r49.ebuild
rename to project-lakitu/app-admin/extensions-manager/extensions-manager-0.0.1-r50.ebuild
diff --git a/project-lakitu/app-admin/extensions-manager/files/cos-extensions.sh b/project-lakitu/app-admin/extensions-manager/files/cos-extensions.sh
index 2ba3117..7e4ade1 100755
--- a/project-lakitu/app-admin/extensions-manager/files/cos-extensions.sh
+++ b/project-lakitu/app-admin/extensions-manager/files/cos-extensions.sh
@@ -15,7 +15,7 @@
 readonly OS_RELEASE="/etc/os-release"
 readonly EXTENSIONS_CACHE="/var/lib/cos-extensions"
 readonly ARTIFACT_REGISTRY_REGIONS=("us" "eu" "asia")
-DEFAULT_GPU_INSTALLER="gcr.io/cos-cloud/cos-gpu-installer:v2.3.5"
+DEFAULT_GPU_INSTALLER="gcr.io/cos-cloud/cos-gpu-installer:v2.4.0"
 
 COS_GPU_INSTALLER="${COS_GPU_INSTALLER:-}"
 
@@ -245,8 +245,8 @@
 
 check_arch() {
   arch=$(uname -m)
-  if  [[ ${arch} != "x86_64" ]]; then
-    echo "GPU installation is only supported on X86 for now.
+  if  [[ ${arch} != "x86_64" && "${arch}" != "aarch64" ]]; then
+    echo "GPU installation is only supported on X86 and ARM64 for now.
     Current architecture detected: ${arch}"
     exit 1
   fi