gpu-wait 0.2.0

A package to run commands when GPU resources are available

GPU Wait

A Python package that monitors GPU resources and executes commands when the GPU becomes available.

Installation

pip install gpu-wait

Usage

Command Line Interface

# Run a command when any GPU is available
gpu-wait python train.py

# Wait for specific GPU
gpu-wait -d 0 python train.py

# Customize memory threshold and polling interval
gpu-wait -t 0.8 -i 5 python train.py

# Enable verbose logging
gpu-wait -v python train.py

Python API

from gpu_wait import GPUMonitor, CommandRunner

# Create monitor with custom threshold
monitor = GPUMonitor(memory_threshold=0.8)

# Create command runner
runner = CommandRunner(monitor)

# Run command when GPU is available
runner.run_when_available(["python", "train.py"])

Features

• Monitor single or multiple GPUs
• Customizable memory threshold and polling interval
• Command line interface
• Python API
• Logging support

Requirements

• Python 3.6+
• NVIDIA GPU
• nvidia-ml-py
• psutil
• click

License

MIT License