torchfunc.cuda

This module provides CUDA related functionalities (e.g. resetting it’s state).

torchfunc.cuda.reset() → None

Reset cuda state by emptying cache and collecting IPC.

Calls torch.cuda.empty_cache() and torch.cuda.ipc_collect() consecutively.

Example:

tensor = torch.cuda.FloatTensor(100, 100)
del tensor
torchfunc.cuda.reset() # Now memory is freed