fix: define docker build args for the stage where they are used

Author: plexoos

Dockerfile

@@ -2,12 +2,13 @@
 
 ARG OS=ubuntu22.04
 ARG CUDA_VERSION=12.4.0
+
+FROM nvidia/cuda:${CUDA_VERSION}-devel-${OS} AS base
+
 ARG OPTIX_VERSION=7.7.0
 ARG GEANT4_VERSION=11.3.2
 ARG CMAKE_VERSION=4.1.2
 
-FROM nvidia/cuda:${CUDA_VERSION}-devel-${OS} AS base
-
 ENV DEBIAN_FRONTEND=noninteractive
 
 RUN apt update \