tensorflow / tensorflow/java

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

Aperta
#500 4 commenti 0 reazioni 0 assegnatari Vedi su GitHub

Nessuno ha ancora preso questa issue.

Lingua principale
Java
Stelle
928
Fork
227
Metriche di merge delle PR
Nessuna PR unita negli ultimi 30g

Descrizione

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

Guida per i contributori

Apri la guida per i contributori

Come iniziare

  1. Leggi tutta la issue e poi la guida ai contributi del progetto.
  2. Commenta sulla issue per dire che te ne occupi tu — evita che due persone facciano lo stesso lavoro.
  3. Fai un fork del repository e lavora su un branch.
  4. Apri una pull request che faccia riferimento al numero della issue.

Direzione di ricerca

Inizia riproducendo la build e l’esecuzione del modello di mvn install java-0.5.0 nell’ambiente Jetson indicato, confrontando la versione di cuDNN utilizzata durante la compilazione con la versione runtime. Il lavoro è completato quando il GPU JAR generato esegue correttamente il modello con versioni compatibili di CUDA e cuDNN, oppure quando viene documentata la configurazione di compilazione richiesta.

Scritto dal modello di indicizzazione a partire dal testo della issue.

Valutazione

Stack tecnologico
java, tensorflow
Ambito
machine-learning
Tipo di issue
Bug
Difficoltà
4/5
Tempo stimato
3-5 giorni
Stato di attività
Tranquilla
Chiarezza
Da chiarire
Idoneità per principianti
42/100

Ricevi le nuove issue nella tua casella

Un breve riepilogo di issue GitHub adatte ai principianti.