gpu-sdk

A Rust SDK for managing GPU compute nodes over WebSocket connections.

What it does

Provides a server that lets GPU nodes register themselves, connect via WebSocket, and share system information including GPU specs, CPU details, and resource availability.

Running

cargo run

Server starts on http://127.0.0.1:8000

API

• POST /register - Register a new node, returns WebSocket URL
• GET /node - List all connected nodes
• DELETE /register/{id} - Unregister a node
• WS /ws/{id} - WebSocket connection for registered nodes

Nodes send system info (CPU, memory, GPU details) during registration.