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: