keep-gpu 0.2.0

Keep GPU is a simple CLI app that keeps your GPUs running

Keep GPU

Keep GPU is a simple CLI app that keeps your GPUs running.

• 🧾 License: MIT
• 📚 Documentation: https://keep-gpu.readthedocs.io

---

Features

• Simple command-line interface
• Uses PyTorch and nvidia-smi to monitor and load GPUs
• Easy to extend for your own keep-alive logic

---

TODO ✅

• Add more CLI args (e.g. --gpu-id, --gpu-ids, --gpu-keep-threshold, --gpu-keep-time, --gpu-keep-vram-usage)
• Add documentation
• Add importable Python functions

---

Installation

pip install keep-gpu

Usage

keep-gpu

Specify the interval in microseconds between GPU usage checks (default is 300 seconds):

keep-gpu --interval 100

Specify GPU IDs to run on (default is all available GPUs):

keep-gpu --gpu-ids 0,1,2

Credits

This package was created with Cookiecutter and the audreyr/cookiecutter-pypackage project template.