ewoksid02.utils.gpu

Functions

cupy_available()
get_best_gpu()	Decides the best GPU in terms of memory available
get_device_name()
get_free_memory(device_id)	Retrieves the available memory on a GPU device
log_allocated_gpu_memory()	Log the status of GPU memory
use_best_gpu()	Set the best available GPU for cupy operations.