Skip to content
/ gpustat Public
forked from wookayin/gpustat

📊 A simple command-line script (wrapper for nvidia-smi) for querying and monitoring GPU status

License

Notifications You must be signed in to change notification settings

cjw85/gpustat

 
 

Folders and files

NameName
Last commit message
Last commit date

Latest commit

 

History

78 Commits
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 

Repository files navigation

gpustat

pypi Build Status license

Just less than nvidia-smi?

Screenshot: gpustat -cp

Usage

$ gpustat

Options:

  • --color : Force colored output (even when stdout is not a tty)
  • --no-color : Suppress color (by default, color is automatically enabled)
  • -u, --show-user : Display username of the process owner
  • -c, --show-cmd : Display the process name
  • -p, --show-pid : Display PID of the process
  • --json : JSON Output (Experimental, #10)

Tips

  • To periodically watch, try watch --color -n1.0 gpustat --color (built-in watch support will be added soon).
  • Running nvidia-smi daemon (root privilege required) will make the query much faster.
  • The GPU ID (index) shown by gpustat (and nvidia-smi) is PCI BUS ID, while CUDA differently assigns the fastest GPU with the lowest ID by default. Therefore, in order to make CUDA and gpustat use same GPU index, configure the CUDA_DEVICE_ORDER environment variable to PCI_BUS_ID (before setting CUDA_VISIBLE_DEVICES for your CUDA program): export CUDA_DEVICE_ORDER=PCI_BUS_ID

Quick Installation

Install from PyPI:

pip install gpustat

To install the latest version (master branch) via pip:

pip install git+https://github.com/wookayin/gpustat.git@master

If you don't have root privilege, please try to install on user namespace: pip install --user. Note that from v0.4, gpustat.py is no more a zero-dependency executable. However, in rare cases you'll need a single executable script (legacy), you can also try:

wget https://raw.githubusercontent.com/wookayin/gpustat/v0.3.2/gpustat.py -O ~/.local/bin/gpustat
chmod +x ~/.local/bin/gpustat    # Assuming ~/.local/bin is in your $PATH

License

MIT License

About

📊 A simple command-line script (wrapper for nvidia-smi) for querying and monitoring GPU status

Resources

License

Stars

Watchers

Forks

Packages

No packages published

Languages

  • Python 100.0%