renom.cuda

renom.cuda.set_cuda_active(activate=True)

If True is given, cuda will be activated.

Parameters:activate (bool) – Activation flag.
renom.cuda.is_cuda_active()

Checks whether CUDA is activated.

Returns:True if cuda is active.
renom.cuda.has_cuda()

This method checks cuda libraries are available.

Returns:True if cuda is correctly set.
renom.cuda.release_mem_pool()

This function releases GPU memory pool.