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.

to_device(data, device[, non_blocking])

Move tensor(s) to device.