rockpool.utilities.backend_management

Utility functionality for managing backends

To check a standard backend, use backend_available(). To check a non-standard backend specification, use check_backend().

To build a shim class that raises an error on instantiation, for when a required backend is not available, use missing_backend_shim().

Functions

backend_available(*backend_names)

Report if a backend is available for use

check_backend(backend_name[, ...])

Check if a backend is available, and register it in a list of available backends

check_samna_available()

check_samna_available controls if samna package is "installed" and "usable" The default backend_available() operation cannot correctly identifies the samna availability.

check_torch_cuda_available()

list_backends()

Print a list of computational backends available in this session

missing_backend_error(class_name, backend_name)

Raise a ModuleNotFoundError exception, with information about a missing backend

missing_backend_shim(class_name, backend_name)

Make a class constructor that raises an error about a missing backend

torch_version_satisfied([req_major, ...])

Check if the installed version of torch satisfies a minimum version requirement

Exceptions

AbortImport