Skip to content

Monitor nvidia gpu unit load associated with docker spawned processes

License

Notifications You must be signed in to change notification settings

AllenCellModeling/nvidia-docker-stats

Folders and files

NameName
Last commit message
Last commit date

Latest commit

 

History

8 Commits
 
 
 
 
 
 
 
 
 
 

Repository files navigation

About

A Python 3 based command line tool for determining a docker container associated processes gpu utilization by joining information across 'nvidia-smi *' and 'docker *' commands.

Usage

Case 1

Clone into the python path and run module as script via:

$ python3 -m nvidiadockerstats.nvidiadockerstats

Case 2

Clone anywhere on network home, link, and run directly:

$ git clone https://github.com/cdw/nvidia-docker-stats.git
$ ln nvidia-docker-stats/nvidiadockerstats/nvidiadockerstats.py ~/bin/nvidiadockerstats
$ nvidiadockerstats

Example output

Container       Image                   pid     gpu_uuid        used_memory     used_gpu
9afcd2624a5b    shiva/keras
					30920   0                11207 MiB       0 %
					30920   1                11587 MiB       0 %
					30920   2                11587 MiB       0 %
					30920   3                11251 MiB       0 %
5eca98f0fa0f    vishnu/pytorch_ext
					8846    0                401 MiB         0 %
					8846    3                353 MiB         0 %

License

GPL 3

About

Monitor nvidia gpu unit load associated with docker spawned processes

Resources

License

Stars

Watchers

Forks

Releases

No releases published

Packages

No packages published

Languages