Tracks computational resources of a process and its child processes, most prominently GPU RAM, as well as RAM and compute time.
Project description
Description
The gpu_tracker package provides a Tracker class that tracks (profiles) the usage of compute time, maximum RAM, and maximum GPU RAM. The compute time is a measurement of the real time taken by the task as opposed to the CPU-utilization time. The GPU tracking is for Nvidia GPUs and uses the nvidia-smi command, assuming the Nvidia drivers have been installed. Computational resources are tracked throughout the duration of a context manager or the duration of explicit calls to the start() and stop() methods of the Tracker class.
Documentation
The complete documentation for the gpu_tracker package, including tutorials, can be found here.
Installation
Requires python 3.10 and above.
Install on Linux, Mac OS X
python3 -m pip install gpu-tracker
Install on Windows
py -3 -m pip install gpu-tracker
PyPi
See our PyPi page here.
Questions, Feature Requests, and Bug Reports
Please submit any questions or feature requests you may have and report any potential bugs/errors you observe on our GitHub issues page.
GitHub Repository
Code is available on GitHub: https://github.com/MoseleyBioinformaticsLab/gpu_tracker.
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
Built Distribution
Hashes for gpu_tracker-1.0.1-py3-none-any.whl
Algorithm | Hash digest | |
---|---|---|
SHA256 | 260c7bb939c6d4ae95cfcce365aa8ad9aa3f736597d7176f9c9d84a4d85bec64 |
|
MD5 | aaa0196e587ec2dc13fdfff5c2f95c90 |
|
BLAKE2b-256 | f9cd7bf4ed89393d349a3ca4095b0da987257c223a97cec6c483d4a03c151cb1 |