PyPI page
Home page
Author:
Andrew H. Bond
Summary:
GPU memory probing and thermal-aware CPU thread control. Works with PyTorch, CuPy, JAX, or any GPU framework.
Latest version:
0.4.0
Required dependencies:
numpy
Optional dependencies:
cupy-cuda12x
|
numpy
|
pytest
|
pytest-cov
|
ruff
|
torch
Downloads last day:
9
Downloads last week:
26
Downloads last month:
101