Support ARM64 & update to RAPIDS 26.04, CUDA 12.9/13.1, Ubuntu 24.04

.devcontainer/cuda12.9/devcontainer.json

@@ -0,0 +1,56 @@
+{
+  "shutdownAction": "stopCompose",
+  "dockerComposeFile": "${localWorkspaceFolder}/docker-compose.devel.yml",
+  "service": "main",
+  "runServices": ["main"],
+  "hostRequirements": {"gpu": "optional"},
+  "features": {
+    "ghcr.io/devcontainers/features/github-cli:1.0.12": {},
+    "ghcr.io/rapidsai/devcontainers/features/utils:latest": {}
+  },
+  "overrideFeatureInstallOrder": [
+    "ghcr.io/devcontainers/features/github-cli",
+    "ghcr.io/rapidsai/devcontainers/features/utils:latest"
+  ],
+  "initializeCommand": ["/bin/bash", "-c", "mkdir -m 0755 -p ${localWorkspaceFolder}/../.{aws,cache,config,log/devcontainer-utils}"],
+  "postAttachCommand": ["/bin/bash", "-c", "if [ ${CODESPACES:-false} = 'true' ]; then . devcontainer-utils-post-attach-command; fi"],
+  "workspaceFolder": "/opt/rapids/node",
+  "mounts": [
+    "source=/etc/timezone,target=/etc/timezone,type=bind",
+    "source=/etc/localtime,target=/etc/localtime,type=bind",
+    "source=${localWorkspaceFolder}/../.aws,target=/opt/rapids/.aws,type=bind",
+    "source=${localWorkspaceFolder}/../.cache,target=/opt/rapids/.cache,type=bind",
+    "source=${localWorkspaceFolder}/../.config,target=/opt/rapids/.config,type=bind",
+    "source=${localWorkspaceFolder}/../.log/devcontainer-utils,target=/var/log/devcontainer-utils,type=bind"
+  ],
+  "customizations": {
+    "vscode": {
+      "extensions": [
+        "augustocdias.tasks-shell-input",
+        "nvidia.nsight-vscode-edition"
+      ],
+      "settings": {
+        "files.watcherExclude": {
+          "**/build/**": true,
+          "**/_skbuild/**": true,
+          "**/target/**": true,
+          "/opt/rapids/.aws/**/*": true,
+          "/opt/rapids/.cache/**/*": true,
+          "/opt/rapids/.conda/**/*": true,
+          "/opt/rapids/.local/share/**/*": true,
+          "/opt/rapids/.vscode-server/**/*": true
+        },
+        "search.exclude": {
+          "**/build/**": true,
+          "**/_skbuild/**": true,
+          "**/*.code-search": true,
+          "/opt/rapids/.aws/**/*": true,
+          "/opt/rapids/.cache/**/*": true,
+          "/opt/rapids/.conda/**/*": true,
+          "/opt/rapids/.local/share/**/*": true,
+          "/opt/rapids/.vscode-server/**/*": true
+        }
+      }
+    }
+  }
+}

.devcontainer/cuda13.1/devcontainer.json

@@ -0,0 +1,56 @@
+{
+  "shutdownAction": "stopCompose",
+  "dockerComposeFile": "${localWorkspaceFolder}/docker-compose.devel.yml",
+  "service": "main",
+  "runServices": ["main"],
+  "hostRequirements": {"gpu": "optional"},
+  "features": {
+    "ghcr.io/devcontainers/features/github-cli:1.0.12": {},
+    "ghcr.io/rapidsai/devcontainers/features/utils:latest": {}
+  },
+  "overrideFeatureInstallOrder": [
+    "ghcr.io/devcontainers/features/github-cli",
+    "ghcr.io/rapidsai/devcontainers/features/utils:latest"
+  ],
+  "initializeCommand": ["/bin/bash", "-c", "mkdir -m 0755 -p ${localWorkspaceFolder}/../.{aws,cache,config,log/devcontainer-utils}"],
+  "postAttachCommand": ["/bin/bash", "-c", "if [ ${CODESPACES:-false} = 'true' ]; then . devcontainer-utils-post-attach-command; fi"],
+  "workspaceFolder": "/opt/rapids/node",
+  "mounts": [
+    "source=/etc/timezone,target=/etc/timezone,type=bind",
+    "source=/etc/localtime,target=/etc/localtime,type=bind",
+    "source=${localWorkspaceFolder}/../.aws,target=/opt/rapids/.aws,type=bind",
+    "source=${localWorkspaceFolder}/../.cache,target=/opt/rapids/.cache,type=bind",
+    "source=${localWorkspaceFolder}/../.config,target=/opt/rapids/.config,type=bind",
+    "source=${localWorkspaceFolder}/../.log/devcontainer-utils,target=/var/log/devcontainer-utils,type=bind"
+  ],
+  "customizations": {
+    "vscode": {
+      "extensions": [
+        "augustocdias.tasks-shell-input",
+        "nvidia.nsight-vscode-edition"
+      ],
+      "settings": {
+        "files.watcherExclude": {
+          "**/build/**": true,
+          "**/_skbuild/**": true,
+          "**/target/**": true,
+          "/opt/rapids/.aws/**/*": true,
+          "/opt/rapids/.cache/**/*": true,
+          "/opt/rapids/.conda/**/*": true,
+          "/opt/rapids/.local/share/**/*": true,
+          "/opt/rapids/.vscode-server/**/*": true
+        },
+        "search.exclude": {
+          "**/build/**": true,
+          "**/_skbuild/**": true,
+          "**/*.code-search": true,
+          "/opt/rapids/.aws/**/*": true,
+          "/opt/rapids/.cache/**/*": true,
+          "/opt/rapids/.conda/**/*": true,
+          "/opt/rapids/.local/share/**/*": true,
+          "/opt/rapids/.vscode-server/**/*": true
+        }
+      }
+    }
+  }
+}

.env.sample

@@ -7,20 +7,16 @@ UID=1000
 PARALLEL_LEVEL=4
 
 # `nvidia/cuda` base image CUDA version
-# CUDA_VERSION=11.6.2
+CUDA_VERSION=13.1.1
 
 # `nvidia/cuda` base image Ubuntu version
-# LINUX_VERSION=ubuntu20.04
+LINUX_VERSION=ubuntu24.04
 
 # List of CUDA device architectures to target.
 # CUDAARCHS=60-real;70-real;75-real;80-real;86
 #
 # Or set to `ALL` to compile for all supported.
-# CUDAARCHS=ALL
-
-# How long sccache should wait until it considers a compile job timed out.
-# This number should be large, because C++ and CUDA can take a long time to compile.
-SCCACHE_IDLE_TIMEOUT=32768
+CUDAARCHS=ALL
 
 # Optional S3 region and bucket for shared sccache.
 #

.github/actions/build-and-publish-image/action.yml

@@ -58,27 +58,27 @@ runs:
   using: composite
   steps:
     - name: Set up QEMU
-      uses: docker/setup-qemu-action@v1
+      uses: docker/setup-qemu-action@v4
     - name: Set up Docker Buildx context
       shell: bash
       run: |
         docker context create builders
     - name: Set up Docker Buildx
-      uses: docker/setup-buildx-action@v2
+      uses: docker/setup-buildx-action@v4
       with:
         endpoint: builders
         buildkitd-flags: --debug
         driver-opts: ${{ inputs.buildx-driver-opts }}
     - name: Login to container registry
       if: inputs.push == 'true'
-      uses: docker/login-action@v2
+      uses: docker/login-action@v4
       with:
         registry: ${{ inputs.registry-url }}
         username: ${{ inputs.registry-username }}
         password: ${{ inputs.registry-password }}
     - name: Build image
       id: docker-build
-      uses: docker/build-push-action@v2
+      uses: docker/build-push-action@v7
       with:
         pull: ${{ inputs.pull }}
         push: ${{ inputs.push }}