TensorPlay
API symbolscuda
Copy
View MarkdownDownload .md

tensorplay.cuda.mem_get_info

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

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

Ask DeepWiki