zero.hardware

Tools related to devices, memory, etc.

functions

free_memory()

Free GPU-memory occupied by torch and run the garbage collector.

get_gpu_info([precise])

Get statistics about GPU devices.