sporco.cuda package¶

This subpackage allows the SPORCO-CUDA extension package to be accessed within the sporco namespace, i.e.

import sporco.cuda

instead of

import sporco_cuda

The import of sporco.cuda will succeed even if the sporco-cuda extension package is not installed, but the availability of these extensions can be determined by checking the boolean value of sporco.cuda.have_cuda. In addition, the function sporco.cuda.device_count() is available independent of whether the import succeeds, allowing it to be used as a reliable test for whether it is possible to run the optimisation functions from sporco-cuda, since this requires both that sporco-cuda be installed and that the value returned by sporco.cuda.device_count() is greater than zero. For example

from sporco import cuda
from sporco.admm import cbpdn

# ...
# Load dictionary D and test image s (and highpass filter it) here
# ...

lmbda = 1e-2
opt = cbpdn.ConvBPDN.Options({'MaxMainIter': 250'})
if cuda.device_count() > 0:
    X = cuda.cbpdn(D, sh, lmbda, opt)
else:
    c = cbpdn.ConvBPDN(D, sh, lmbda, opt)
    X = c.solve()

The content of the sporco.cuda namespace is summarised below. For full details of the functions listed here, see the SPORCO-CUDA documentation.

Always available¶

sporco.cuda.have_cuda¶: A boolean value indicating whether the import of sporco_cuda succeeded.

sporco.cuda.device_count()¶

Get the number of CUDA GPU devices installed on the host system. Returns 0 if no devices are installed or if the import of sporco_cuda failed.

Returns

ndevint: Number of installed devices

Only available if `have_cuda` is True¶

sporco.cuda.current_device(id=None)¶

Get or set the current CUDA GPU device. The current device is not set if id is None.

Parameters

idint or None, optional (default None): Device number of device to be set as current device

Returns

idint: Device number of current device

sporco.cuda.memory_info()¶

Get memory information for the current CUDA GPU device.

Returns

freeint: Free memory in bytes
totalint: Total memory in bytes

sporco.cuda.device_name(int dev=0)¶

Get hardware model name for the specified CUDA GPU device.

Parameters

idint, optional (default 0): Device number of device

Returns

namestring: Hardware device name

sporco.cuda.cbpdn(D, S, lmbda, opt, dev=0)¶

A GPU-accelerated version of admm.cbpdn.ConvBPDN. Multiple images and multi-channel images in input signal S are currently not supported.

A usage example is available.

Parameters

Darray_like(float32, ndim=3): Dictionary array (three dimensional)
Sarray_like(ndim=2): Signal array (two dimensional)
lmbdafloat32: Regularisation parameter
optdict or admm.cbpdn.ConvBPDN.Options object: Algorithm options
devint: Device number of GPU device to use

Returns

Xndarray: Coefficient map array (sparse representation)

sporco.cuda.cbpdngrd(D, S, lmbda, mu, opt, dev=0)¶

A GPU-accelerated version of admm.cbpdn.ConvBPDNGradReg. Multiple images and multi-channel images in input signal S are currently not supported.

A usage example is available.

Parameters

Darray_like(float32, ndim=3): Dictionary array (three dimensional)
Sarray_like(ndim=2): Signal array (two dimensional)
lmbdafloat32: Regularisation parameter (\(\ell_1\))
mufloat: Regularisation parameter (\(\ell_2\) of gradient)
optdict or admm.cbpdn.ConvBPDNGradReg.Options object: Algorithm options
devint: Device number of GPU device to use

Returns

Xndarray: Coefficient map array (sparse representation)

sporco.cuda.cbpdnmsk(D, s, w, lmbda, opt, dev=0)¶

A GPU-accelerated version of admm.cbpdn.AddMaskSim used together with admm.cbpdn.ConvBPDN, providing a spatial mask in the data fidelity term of the functional minimized by this class. Multiple images and multi-channel images in input signal S are currently not supported.

Since the spatial mask is implemented via the Additive Mask Simulation (AMS) method [50], the entries must be in \(\{0,1\}\). Note that this GPU version differs from the Python code in its handling of the L1Weight option: this version automatically adjusts this array to account for the AMS impulse filter that is inserted into the dictionary, while the Python version requires this to be handled by the calling function. In addition, this version prepends the AMS impulse filter at the start of the dictionary, while the Python version appends it at the end.

A usage example is available.

Parameters

Darray_like(float32, ndim=3): Dictionary array (three dimensional)
sarray_like(float32, ndim=2): Signal array (two dimensional)
warray_like: Mask array (two dimensional)
lmbdafloat32: Regularisation parameter
optdict or admm.cbpdn.ConvBPDN.Options object: Algorithm options
devint: Device number of GPU device to use

Returns

Xndarray: Coefficient map array (sparse representation)

sporco.cuda.cbpdngrdmsk(D, s, w, lmbda, mu, opt, dev=0)¶

A GPU-accelerated version of of admm.cbpdn.AddMaskSim used together with admm.cbpdn.ConvBPDNGradReg, providing a spatial mask in the data fidelity term of the functional minimized by this class. Multiple images and multi-channel images in input signal S are currently not supported.

Since the spatial mask is implemented via the Additive Mask Simulation (AMS) method [50], the entries must be in \(\{0,1\}\). Note that this GPU version differs from the Python code in its handling of the L1Weight and GradWeight options: this version automatically adjusts these arrays to account for the AMS impulse filter that is inserted into the dictionary, while the Python version requires this to be handled by the calling function. In addition, this version prepends the AMS impulse filter at the start of the dictionary, while the Python version appends it at the end.

A usage example is available.

Parameters

Darray_like(float32, ndim=3): Dictionary array (three dimensional)
sarray_like(float32, ndim=2): Signal array (two dimensional)
warray_like: Mask array (two dimensional)
lmbdafloat32: Regularisation parameter (\(\ell_1\))
mufloat: Regularisation parameter (\(\ell_2\) of gradient)
optdict or admm.cbpdn.ConvBPDNGradReg.Options object: Algorithm options
devint: Device number of GPU device to use

Returns

Xndarray: Coefficient map array (sparse representation)

sporco.cuda package¶

Always available¶

Only available if have_cuda is True¶

Only available if `have_cuda` is True¶