TensorPlay
API symbolscuda
Copy
View MarkdownDownload .md

tensorplay.cuda.memory.caching_allocator_disabled

tensorplay.cuda.memory.caching_allocator_disabled()[source]

Context manager that temporarily disables the CUDA caching allocator.

tensorplay.cuda.memory_usage

Return the percent of time over the past sample period during which global (device) memory was being read or written as given by nvidia-smi .

tensorplay.cuda.memory.caching_allocator_enable

Enable or disable the CUDA memory allocator. On by default.

Ask DeepWiki