Skip to main content

A quick access to nvidia gpu information

Project description

# gpuinfo

I implement some functions that can help users to obtain nvidia gpu information.

To use gpuinfo, you need to be able to run 'ps' and 'nvidia-smi' in your terminal.

# Install with pip
pip install gpuinfo
I only tested on linux system with python3.

# Usage

from gpuinfo import GPUInfo

GPUInfo has the following functions:

return a dict. show every user and memory on a certain gpu

return a list containing all GPU ids that no process is using currently.

return a dict and three lists. pid_list has pids as keys and gpu ids as values, showing which gpu the process is using

Input a pid number , return its creator by linux command ps

return two lists. The first list contains usage percent of every GPU. The second list contains the memory used of every GPU. The information is obtained by command 'nvidia-smi'

# Example

from gpuinfo import GPUInfo

min_percent=percent.index(min([percent[i] for i in available_device]))
min_memory=memory.index(min([memory[i] for i in available_device]))

torch.cuda.set_device(min_percent) 或者(min_memory)

Project details

Download files

Download the file for your platform. If you're not sure which to choose, learn more about installing packages.

Filename, size & hash SHA256 hash help File type Python version Upload date
gpuinfo-1.0.0a4-py3-none-any.whl (3.3 kB) Copy SHA256 hash SHA256 Wheel py3
gpuinfo-1.0.0a4.tar.gz (5.2 kB) Copy SHA256 hash SHA256 Source None

Supported by

Elastic Elastic Search Pingdom Pingdom Monitoring Google Google BigQuery Sentry Sentry Error logging AWS AWS Cloud computing DataDog DataDog Monitoring Fastly Fastly CDN SignalFx SignalFx Supporter DigiCert DigiCert EV certificate StatusPage StatusPage Status page