mindspore.device_context.gpu.is_available

View Source On Gitee
mindspore.device_context.gpu.is_available()[source]

Return a bool indicating if CUDA is currently available.

Returns

Bool, indicating if CUDA is currently available.

Examples

>>> import mindspore as ms
>>> ms.device_context.gpu.device.is_available()