Metadata-Version: 2.4
Name: gpu-worker
Version: 0.1.2
Summary: GPUBridge Worker — registers with a GPUBridge controller, polls for jobs, and executes them in Docker containers with optional GPU passthrough.
Author-email: Amruth Vamshi <battaamruth1@gmail.com>
License-Expression: MIT
Project-URL: Homepage, https://github.com/LIU-CS-691/gpu-bridge
Project-URL: Repository, https://github.com/LIU-CS-691/gpu-bridge
Project-URL: Issues, https://github.com/LIU-CS-691/gpu-bridge/issues
Classifier: Development Status :: 3 - Alpha
Classifier: Environment :: Console
Classifier: Intended Audience :: Developers
Classifier: Topic :: Scientific/Engineering :: Artificial Intelligence
Classifier: Programming Language :: Python :: 3.11
Classifier: Programming Language :: Python :: 3.12
Requires-Python: >=3.11
Description-Content-Type: text/markdown
Requires-Dist: typer>=0.12.5
Requires-Dist: httpx>=0.27.2
Requires-Dist: python-dotenv>=1.0.0
Requires-Dist: docker>=7.0.0
Requires-Dist: nvidia-ml-py>=12.0.0

# gpu-worker

Worker agent for [GPUBridge](https://github.com/LIU-CS-691/gpu-bridge) — registers with a GPUBridge controller, polls for jobs, and executes them in Docker containers with optional NVIDIA GPU passthrough.

## Install

```bash
pip install gpu-worker
```

## Quick Start

```bash
# 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)
