From 70c3695f8d41c3d1e50d34ff1ad81a3bfcafe8f5 Mon Sep 17 00:00:00 2001
From: MyronRodrigues-StreetDrone <myron.rodrigues@streetdrone.com>
Date: Mon, 29 Jan 2024 16:57:23 +0530
Subject: [PATCH] fix: kernel headers only if balena version gt 2

---
 gpu/Dockerfile | 5 +++--
 1 file changed, 3 insertions(+), 2 deletions(-)

diff --git a/gpu/Dockerfile b/gpu/Dockerfile
index 9fe4570..1e4e602 100644
--- a/gpu/Dockerfile
+++ b/gpu/Dockerfile
@@ -27,8 +27,9 @@ SHELL ["/bin/bash", "-o", "pipefail", "-c"]
 
 # Download the kernel source then prepare kernel source to build a module.
 RUN \
-    curl -fsSL "https://files.balena-cloud.com/images/${BALENA_MACHINE_NAME}/${VERSION}/kernel_source.tar.gz" \
-        | tar xz --strip-components=2 && \
+    [ "${VERSION%%.*}" -gt "2" ] && FILE_NAME="kernel_modules_headers.tar.gz" || FILE_NAME="kernel_source.tar.gz" && \
+    curl -fsSL "https://files.balena-cloud.com/images/${BALENA_MACHINE_NAME}/${VERSION}/${FILE_NAME}" \
+    | tar xz --strip-components=2 && \
     make -C build modules_prepare -j"$(nproc)"
 
 # required if using install-libglvnd from nvidia-installer below