tensorplay.cuda.mem_get_info

tensorplay.cuda.mem_get_info(device: Any = None) → tuple[int, int][source]

Return the global free and total GPU memory for a given device using cudaMemGetInfo.