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.
https://pypi.org/project/gpuinfo/

# Usage

```python
from gpuinfo import GPUInfo
```

GPUInfo has the following functions:

get_users(gpu_id)
return a dict. show every user and memory on a certain gpu

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

get_info()
pid_list,percent,memory,gpu_used=get_info()
return a dict and three lists. pid_list has pids as keys and gpu ids as values, showing which gpu the process is using

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

gpu_usage()
gpu_usage()
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

```python
from gpuinfo import GPUInfo

available_device=GPUInfo.check_empty()
#available_device就是一个含有所有没有任务的gpu编号的列表
percent,memory=GPUInfo.gpu_usage()
#获得所有gpu的使用百分比和显存占用量
min_percent=percent.index(min([percent[i] for i in available_device]))
#未被使用的gpu里percent最小的
min_memory=memory.index(min([memory[i] for i in available_device]))
#未被使用的gpu里显存占用量最少的

#如果你使用pytorch
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.

Source Distribution

gpuinfo-1.0.0a7.tar.gz (5.3 kB view details)

Uploaded Source

Built Distribution

gpuinfo-1.0.0a7-py3-none-any.whl (4.1 kB view details)

Uploaded Python 3

File details

Details for the file gpuinfo-1.0.0a7.tar.gz.

File metadata

  • Download URL: gpuinfo-1.0.0a7.tar.gz
  • Upload date:
  • Size: 5.3 kB
  • Tags: Source
  • Uploaded using Trusted Publishing? No
  • Uploaded via: twine/1.13.0 pkginfo/1.4.2 requests/2.19.1 setuptools/41.4.0 requests-toolbelt/0.9.1 tqdm/4.26.0 CPython/3.7.4

File hashes

Hashes for gpuinfo-1.0.0a7.tar.gz
Algorithm Hash digest
SHA256 56377301dbbd09c22b1bc19d5bb5d291b835676ed8e95f9004c195a091c7a077
MD5 f10418f11882f80b75dfb10bcfa0b051
BLAKE2b-256 03940d99b8b788cd770cbd85f9324d874208c350f6c6d91596374a612ae558d3

See more details on using hashes here.

File details

Details for the file gpuinfo-1.0.0a7-py3-none-any.whl.

File metadata

  • Download URL: gpuinfo-1.0.0a7-py3-none-any.whl
  • Upload date:
  • Size: 4.1 kB
  • Tags: Python 3
  • Uploaded using Trusted Publishing? No
  • Uploaded via: twine/1.13.0 pkginfo/1.4.2 requests/2.19.1 setuptools/41.4.0 requests-toolbelt/0.9.1 tqdm/4.26.0 CPython/3.7.4

File hashes

Hashes for gpuinfo-1.0.0a7-py3-none-any.whl
Algorithm Hash digest
SHA256 a0dc7191542c95599e807ae103e80d63138fa82a472d0b7d22e757208586e6ab
MD5 5e1b5a4d6bc2ff8d9cfdab99605d3e1f
BLAKE2b-256 64c8bb3e935843f583fbf9970e125b888a2c7c0ec6ce6eada8d3f69d88be0294

See more details on using hashes here.

Supported by

AWS Cloud computing and Security Sponsor Datadog Monitoring Fastly CDN Google Download Analytics Pingdom Monitoring Sentry Error logging StatusPage Status page