fix: reenable cuda builds for 24.04

Greenroom-Robotics · Aug 7, 2024 · fe9cddd · fe9cddd
1 parent 6353b0a
commit fe9cddd
Showing 1 changed file with 11 additions and 11 deletions.
diff --git a/scripts/docker-build.py b/scripts/docker-build.py
@@ -7,7 +7,7 @@
 
 UBUNTU_VERSION = "24.04"
 UBUNTU_CODENAME = "noble"
-CUDA_VERSION = "11.7.1-devel-ubuntu22.04"
+CUDA_VERSION = f"12.5.1-devel-ubuntu{UBUNTU_VERSION}"
 
 ENV = Dict[str, str]
 
@@ -47,16 +47,16 @@ def build_image(base_image: str, ros_distro: str, arch: str, tags: List[str], pu
 
 
     # Build images
-    # build_image(
-    #     base_image=f"nvidia/cuda:{CUDA_VERSION}",
-    #     ros_distro=args.ros_distro,
-    #     arch=args.arch,
-    #     tags=[
-    #         f"ghcr.io/greenroom-robotics/ros_builder:{args.ros_distro}-{args.version}-cuda-{args.arch}",
-    #         f"ghcr.io/greenroom-robotics/ros_builder:{args.ros_distro}-latest-cuda-{args.arch}"
-    #     ],
-    #     push=args.push,
-    # )
+    build_image(
+        base_image=f"nvidia/cuda:{CUDA_VERSION}",
+        ros_distro=args.ros_distro,
+        arch=args.arch,
+        tags=[
+            f"ghcr.io/greenroom-robotics/ros_builder:{args.ros_distro}-{args.version}-cuda-{args.arch}",
+            f"ghcr.io/greenroom-robotics/ros_builder:{args.ros_distro}-latest-cuda-{args.arch}"
+        ],
+        push=args.push,
+    )
     build_image(
         base_image=f"ubuntu:{UBUNTU_CODENAME}",
         ros_distro=args.ros_distro,