Skip to content

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

Open
@rctay

Description

@rctay

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!

Metadata

Metadata

Assignees

No one assigned

    Labels

    No labels
    No labels

    Type

    No type

    Projects

    No projects

    Milestone

    No milestone

    Relationships

    None yet

    Development

    No branches or pull requests

    Issue actions