diff --git a/.github/workflows/build-release.yml b/.github/workflows/build-release.yml
index a9ad7ae..44eecd6 100644
--- a/.github/workflows/build-release.yml
+++ b/.github/workflows/build-release.yml
@@ -20,7 +20,10 @@ on:
 jobs:
   build-linux-x86_64:
     runs-on: ubuntu-22.04
+    container: nvidia/cuda:12.6.2-devel-ubuntu22.04
     steps:
+      - name: Upgrade git
+        run: apt-get update && apt-get install -y git
       - uses: actions/checkout@v4
         with:
           submodules: "true"
@@ -55,7 +58,10 @@ jobs:
 
   build-linux-arm64:
     runs-on: ubuntu-22.04-arm
+    container: nvidia/cuda:12.6.2-devel-ubuntu22.04
     steps:
+      - name: Upgrade git
+        run: apt-get update && apt-get install -y git
       - uses: actions/checkout@v4
         with:
           submodules: "true"
@@ -72,7 +78,8 @@ jobs:
       - uses: actions/setup-node@v4.0.2
         with:
           node-version: 20
-          cache: "yarn"
+      - name: Install yarn
+        run: npm install -g yarn
       - name: Install dependencies
         run: yarn install
       - name: Prepare & build
diff --git a/scripts/prepare-linux.sh b/scripts/prepare-linux.sh
index 4702306..24f8e7d 100755
--- a/scripts/prepare-linux.sh
+++ b/scripts/prepare-linux.sh
@@ -14,12 +14,15 @@ export DEBIAN_FRONTEND=noninteractive
 
 ARCH=${ARCH:-${1:-$(uname -m)}}
 
+run_as_root apt-get update
+run_as_root apt-get install -qy lsb-release wget
+
 if [ $ARCH == "x86_64" ]; then
   DISTRO=$(lsb_release -c -s)
   wget -qO- https://packages.lunarg.com/lunarg-signing-key-pub.asc | run_as_root tee /etc/apt/trusted.gpg.d/lunarg.asc
   run_as_root wget -qO /etc/apt/sources.list.d/lunarg-vulkan-1.3.280-$DISTRO.list https://packages.lunarg.com/vulkan/1.3.280/lunarg-vulkan-1.3.280-$DISTRO.list
   run_as_root apt-get update
-  run_as_root apt-get install -qy vulkan-sdk
+  run_as_root apt-get install -qy vulkan-sdk cmake pkg-config build-essential  libx11-xcb-dev libxkbcommon-dev libwayland-dev libxrandr-dev
 else
   run_as_root apt-get update
   run_as_root apt-get install -qy curl gnupg2