Skip to content

A simple script (wrapper for nvidia-smi) for monitoring GPU status across cluster

License

Notifications You must be signed in to change notification settings

manzilzaheer/gpustat

 
 

Folders and files

NameName
Last commit message
Last commit date

Latest commit

 

History

51 Commits
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 

Repository files navigation

clustergpustat

license

Just nvidia-smi across the cluster

Usage

  1. $ gpustat it lists all gpu statistics across the cluster
  2. $ gpufree it lists all gpu free across the cluster
  3. sshfree logins to a random node with a free gpu

Quick Installation

Just download gpustat, and/or gpufree, and/or gpuonefree and sshfree into somewhere in PATH, e.g. ~/.local/bin/ (when you do not have root privilege, for example):

sudo wget https://git.io/gpustat -O /usr/local/bin/gpustat && sudo chmod +x /usr/local/bin/gpustat
sudo wget https://git.io/gpufree -O /usr/local/bin/gpufree && sudo chmod +x /usr/local/bin/gpufree
sudo wget https://git.io/gpuonefree -O /usr/local/bin/gpuonefree && sudo chmod +x /usr/local/bin/gpuonefree
sudo wget https://git.io/sshfree -O /usr/local/bin/gpustat && sudo chmod +x /usr/local/bin/sshfree

License

MIT License

About

A simple script (wrapper for nvidia-smi) for monitoring GPU status across cluster

Resources

License

Stars

Watchers

Forks

Packages

No packages published

Languages

  • Python 100.0%