PytorchBackend.get_available_memory#

PytorchBackend.get_available_memory()[source]#

Returns the available memory available for computations in bytes. For CPU operations this corresponds to available RAM. For GPU operations the function is expected to return the available GPU memory.