gpu_is_available

ivy.gpu_is_available(f=None)[source]

Determine whether a GPU is available to use, with the backend framework.

Parameters

f (ml_framework, optional) – Machine learning framework. Inferred from inputs if None.

Returns

Boolean, as to whether a gpu is available.


Supported Frameworks:

empty jax_logo empty tf_logo empty pytorch_logo empty mxnet_logo empty numpy_logo empty