From 4ad8a8793e6c98238cd9570b3d61637df13ae035 Mon Sep 17 00:00:00 2001
From: jaakko <jaakko@kipina.tech>
Date: Mon, 6 Apr 2026 21:02:43 +0300
Subject: [PATCH] Native node CUDA Docker: nvidia/cuda base + GPU runtime
MIME-Version: 1.0
Content-Type: text/plain; charset=UTF-8
Content-Transfer-Encoding: 8bit

Dockerfile käyttää nvidia/cuda:12.6.3 -imagea jossa CUDA-kirjastot
ovat valmiina. docker-compose lisää runtime: nvidia + NVIDIA_VISIBLE_DEVICES.

Co-Authored-By: Claude Opus 4.6 (1M context) <noreply@anthropic.com>
---
 network-poc/Dockerfile.native-node | 14 +++++++++-----
 network-poc/docker-compose.yml     |  3 ++-
 2 files changed, 11 insertions(+), 6 deletions(-)

diff --git a/network-poc/Dockerfile.native-node b/network-poc/Dockerfile.native-node
index 2329389..d10f704 100644
--- a/network-poc/Dockerfile.native-node
+++ b/network-poc/Dockerfile.native-node
@@ -1,9 +1,13 @@
-FROM rust:slim AS builder
+FROM nvidia/cuda:12.6.3-devel-ubuntu24.04 AS builder
 
 RUN apt-get update && apt-get install -y \
-    pkg-config libssl-dev g++ \
+    curl pkg-config libssl-dev g++ libvulkan-dev \
     && rm -rf /var/lib/apt/lists/*
 
+# Rust
+RUN curl --proto '=https' --tlsv1.2 -sSf https://sh.rustup.rs | sh -s -- -y
+ENV PATH="/root/.cargo/bin:${PATH}"
+
 WORKDIR /app
 COPY Cargo.toml Cargo.lock ./
 COPY hub/Cargo.toml hub/Cargo.toml
@@ -20,11 +24,11 @@ RUN mkdir -p hub/src node/src native-node/src \
 COPY native-node/src native-node/src
 RUN cargo build --release -p native-node
 
-FROM debian:bookworm-slim
-RUN apt-get update && apt-get install -y ca-certificates && rm -rf /var/lib/apt/lists/*
+FROM nvidia/cuda:12.6.3-runtime-ubuntu24.04
+RUN apt-get update && apt-get install -y ca-certificates libvulkan1 && rm -rf /var/lib/apt/lists/*
 COPY --from=builder /app/target/release/native-node /usr/local/bin/native-node
 
-ENV HUB_URL=ws://hub:3000/ws
+ENV HUB_URL=ws://agentic-poc:3000/ws
 ENV ALLOCATED_GB=4
 
 CMD ["native-node"]
diff --git a/network-poc/docker-compose.yml b/network-poc/docker-compose.yml
index b1ed68c..a7529d9 100644
--- a/network-poc/docker-compose.yml
+++ b/network-poc/docker-compose.yml
@@ -17,12 +17,13 @@ services:
       context: .
       dockerfile: Dockerfile.native-node
     container_name: kipina_native_node
+    runtime: nvidia
     environment:
       - HUB_URL=ws://agentic-poc:3000/ws
       - ALLOCATED_GB=4
+      - NVIDIA_VISIBLE_DEVICES=all
     depends_on:
       - agentic-poc
-    # GPU passthrough (valinnainen — toimii myös ilman)
     deploy:
       resources:
         reservations: