tensorflow / tensorflow/java

Compiling from source, cuDNN version is not compatible? How can I change the cuDNN compile version?

Abierto
#500 4 comentarios 0 reacciones 0 asignados Ver en GitHub

Nadie ha tomado este issue todavía.

Lenguaje dominante
Java
Estrellas
928
Forks
227
Métricas de merge de PR
Sin PR fusionados en 30 d

Descripción

I use mvn install java-0.5.0 in jetson with below environment. Finally it completed and generated the "tensorflow-core-api-0.5.0-linux-arm64-gpu.jar".

System information

  • OS Platform and Distribution : Linux Ubuntu 20.04 arm64
  • TensorFlow installed from (source or binary): source
  • TensorFlow version: 2.10.1
  • Java version (i.e., the output of java -version): openjdk version "11.0.20"
  • Java command line flags (e.g., GC parameters):
  • Installed from Maven Central?:
  • Bazel version (if compiling from source): 5.4.1
  • GCC/Compiler version (if compiling from source): 9.4.0
  • CUDA/cuDNN version: CUDA11.4+CuDNN8.6.0
  • GPU model and memory:

But when I run the model , the error is occured. Here is the error log:
"2023-08-01 09:57:03.793190: E external/org_tensorflow/tensorflow/stream_executor/cuda/cuda_dnn.cc:377] Loaded runtime CuDNN library: 8.6.0 but source was compiled with: 8.9.0. CuDNN library needs to have matching major version and equal or higher minor version. If using a binary install, upgrade your CuDNN library. If building from sources, make sure the library loaded at runtime is compatible with the version specified during compile configuration."
453ad32b83ae0637ea2e646e0504851

I have tried upgrade CuDNN to 8.9.0 and run again, but it's run failed with error-"tensorflow/core/framework/op_kernel.cc:1780] OP_REQUIRES failed at conv_ops.cc:1143 : NOT_FOUND: No algorithm worked! Error messages:". I think this may be the environment doesn't match.
614c484dff4a18d1c0a9bd186efcfa2

So how can I resolve this problem? I think this is a way to fix the problem that is to change the cuDNN compile version, but I can't find any info about this.
P.S.: I run the model successfully in my local machine(windows11 x86_64...)

Best Regards

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.

Línea de trabajo

Comienza reproduciendo la compilación y la ejecución del modelo de mvn install java-0.5.0 en el entorno Jetson indicado, comparando la versión de cuDNN utilizada durante la compilación con la versión en tiempo de ejecución. Se considera completado cuando el JAR de GPU generado ejecuta correctamente el modelo con versiones compatibles de CUDA y cuDNN, o cuando se documenta la configuración de compilación necesaria.

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

Evaluación

Stack tecnológico
java, tensorflow
Área
machine-learning
Tipo de issue
Error
Dificultad
4/5
Tiempo estimado
3-5 días
Estado de actividad
Tranquilo
Claridad
Necesita aclaración
Aptitud para principiantes
42/100

Recibe los nuevos issues en tu correo

Un resumen breve de issues de GitHub para principiantes.