gpu-worker 0.1.0

GPUBridge Worker — registers with a GPUBridge controller, polls for jobs, and executes them in Docker containers with optional GPU passthrough.

gpu-worker

Worker agent for GPUBridge — registers with a GPUBridge controller, polls for jobs, and executes them in Docker containers with optional NVIDIA GPU passthrough.

Install

pip install gpu-worker

Quick Start

# Set your invite token (get one from your admin)
export GPU_BRIDGE_INVITE="<invite_token>"

# Register this machine
gpu-worker register --name "My GPU Server"

# Start executing jobs
gpu-worker start --worker-id <WORKER_ID> --gpu

Requirements

• Python 3.11+
• Docker (running)
• NVIDIA Container Toolkit (for GPU jobs)