v0.1.0 - Initial Release

First public release of gpu-memory-guard.

Features

• Check GPU VRAM usage via nvidia-smi (subprocess) or pynvml
• • Estimate whether a model will fit in available VRAM
• • Configurable safety buffer (default 512MB)
• • CLI interface: gpu-guard --check <size_gb>
• • Python API: from gpu_guard import check_vram

Install

pip install gpu-memory-guard