tensorflow / tensorflow/java

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

オープン
#500 コメント 4 件 リアクション 0 件 担当者 0 名 GitHub で見る

まだ誰も着手していません。

主要言語
Java
スター
928
フォーク
227
PR マージ指標
30日以内にマージされた PR はありません

説明

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

コントリビューションガイド

コントリビューションガイドを開く

はじめの一歩

  1. issue を最後まで読み、次にプロジェクトのコントリビューションガイドを読みます。
  2. 着手することを issue にコメントします — 二人が同じ作業をするのを防げます。
  3. リポジトリをフォークし、ブランチを切って変更します。
  4. issue 番号を参照したプルリクエストを送ります。

調査の方向性

報告された Jetson 環境で mvn install java-0.5.0 のビルドとモデル実行を再現することから始め、コンパイル時に使用された cuDNN バージョンとランタイムバージョンを比較します。生成された GPU JAR が互換性のある CUDA および cuDNN バージョンでモデルを正常に実行できるか、または必要なコンパイル設定が文書化されていれば完了です。

索引モデルが issue の本文から書いたものです。

評価

技術スタック
java, tensorflow
領域
machine-learning
issue の種類
バグ
難易度
4/5
見積もり時間
3〜5日
活発さ
静か
明瞭さ
説明が足りない
初心者へのやさしさ
42/100

新しい issue をメールで受け取る

初心者向けの GitHub issue を短くまとめたダイジェスト。