java icon indicating copy to clipboard operation
java copied to clipboard

Advice needed: how to check number of/any CUDA devices available?

Open rctay opened this issue 6 months ago • 1 comments

This template is for miscellaneous issues not covered by the other issue categories.

Hi, is there a way to check via Tensorflow Java the number of / or any CUDA devices available? For example, in Python with Torch one could do

import torch
torch.cuda.is_available()

Thanks in advance!

rctay avatar Jun 29 '25 13:06 rctay