Rate this Page
★★★★★

CUDAPluggableAllocator#

classtorch.cuda.memory.CUDAPluggableAllocator(path_to_so_file,alloc_fn_name,free_fn_name)[source]#

CUDA memory allocator loaded from a so file.