mcframework.backends.is_cuda_available#

mcframework.backends.is_cuda_available() bool[source]#

Check if CUDA is available.

Returns:
bool

True if CUDA is available and PyTorch was built with CUDA support.

Examples

>>> if is_cuda_available():
...     backend = TorchCUDABackend()