TensorPlay
API symbolscuda
Copy
View MarkdownDownload .md

tensorplay.cuda.memory.use_mem_pool

tensorplay.cuda.memory.use_mem_pool(pool)[source]

Route allocations to a MemPool (not supported by this build).

tensorplay.cuda.memory.caching_allocator_enable

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

tensorplay.cuda.nccl.is_available

Next Page

Ask DeepWiki