Metadata-Version: 2.4
Name: nvidia-mig-py
Version: 0.1.1
Summary: Python bindings for Nvidia MIG
Author-email: Zhiwei Chu <chi32@purdue.edu>
Classifier: Development Status :: 1 - Planning
Classifier: Intended Audience :: Developers
Classifier: Intended Audience :: System Administrators
Classifier: Environment :: GPU :: NVIDIA CUDA :: 12 :: 12.3
Classifier: Operating System :: POSIX :: Linux
Classifier: Programming Language :: Python
Classifier: Topic :: Software Development :: Libraries :: Python Modules
Classifier: Topic :: System :: Hardware
Classifier: Topic :: System :: Systems Administration
Description-Content-Type: text/markdown
License-File: LICENSE
Requires-Dist: nvidia-ml-py>=12.560.30
Dynamic: license-file

# NVIDIA MIG Manager

A Python library for managing NVIDIA GPU Multi-Instance GPU (MIG) partitions and resources. This library provides a simple interface to create, list, and destroy MIG partitions, as well as monitor their resource usage. **Only tested with NVIDIA A30 GPUs.**

---

## Features

- **MIG Partition Management**: Create, list, and destroy MIG partitions on NVIDIA GPUs that support MIG.
- **Resource Allocator**: Automatically create MIG partitions based on workload memory requirements.
- **Resource Monitoring**: Track memory, compute utilization, temperature, and power usage for each MIG partition to maintain optimal performance and prevent bottlenecks.

---

## Installation

### Requirements

- **Python**: >= 3.7
- **NVIDIA GPU with MIG support**
- **NVIDIA Driver and DCGM Python bindings**

### Install via pip

To install the library, simply clone the repository and install dependencies:

```bash
git clone git@github.com:MoonOoOoO/nvidia-mig-py.git
pip install .
```
