Keep GPU is a simple CLI app that keeps your GPUs running.
- 🧾 License: MIT
- 📚 Documentation: https://keepgpu.readthedocs.io
- Simple command-line interface
- Uses PyTorch and
nvidia-smito monitor and load GPUs - Easy to extend for your own keep-alive logic
- 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
pip install keep-gpukeep-gpuSpecify the interval in microseconds between GPU usage checks (default is 300 seconds):
keep-gpu --interval 100Specify GPU IDs to run on (default is all available GPUs):
keep-gpu --gpu-ids 0,1,2This package was created with Cookiecutter and the audreyr/cookiecutter-pypackage project template.