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
Release history Release notifications | RSS feed
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)
Built Distribution
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
Algorithm | Hash digest | |
---|---|---|
SHA256 |
56377301dbbd09c22b1bc19d5bb5d291b835676ed8e95f9004c195a091c7a077
|
|
MD5 |
f10418f11882f80b75dfb10bcfa0b051
|
|
BLAKE2b-256 |
03940d99b8b788cd770cbd85f9324d874208c350f6c6d91596374a612ae558d3
|
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
Algorithm | Hash digest | |
---|---|---|
SHA256 |
a0dc7191542c95599e807ae103e80d63138fa82a472d0b7d22e757208586e6ab
|
|
MD5 |
5e1b5a4d6bc2ff8d9cfdab99605d3e1f
|
|
BLAKE2b-256 |
64c8bb3e935843f583fbf9970e125b888a2c7c0ec6ce6eada8d3f69d88be0294
|