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

Abierto
#614 1 comentario 0 reacciones 0 asignados Ver en GitHub

Nadie ha tomado este issue todavía.

Evaluación

Dificultad
5/5
Tiempo estimado
Más de una semana
Aptitud para principiantes
25/100
Tipo de issue
Nueva funcionalidad
Claridad
Necesita aclaración
Estado de actividad
Estancado
Stack tecnológico
java, tensorflow
Área
machine-learning

Línea de trabajo

No se mencionan archivos Java de TensorFlow, pruebas ni puntos de entrada. Empieza por aclarar si la solicitud se refiere a una API existente para comprobar dispositivos o a una nueva capacidad, tomando el ejemplo de Python Torch como el comportamiento esperado. El trabajo estaría terminado cuando se haya confirmado una API o se hayan definido el alcance de la implementación y el enfoque de validación.

Escrito por el modelo de indexación a partir del texto del issue.

Descripción

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!

Lenguaje dominante
Java
Estrellas
928
Forks
227
Sin PR fusionados en 30 d

Guía de contribución

Abrir la guía de contribución

Primeros pasos

  1. Lee el issue completo y luego la guía de contribución del proyecto.
  2. Comenta en el issue que vas a ocuparte — evita que dos personas hagan lo mismo.
  3. Haz un fork del repositorio y trabaja en una rama.
  4. Abre un pull request que haga referencia al número del issue.

Más de tensorflow/java

Todos los issues de tensorflow/java

Issues similares

Más issues de Java

Recibe los nuevos issues en tu correo

Un resumen breve de issues de GitHub para principiantes.