Compiling from source, cuDNN version is not compatible? How can I change the cuDNN compile version?
还没有人认领这个 Issue。
- 主要语言
- 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."
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.
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
贡献指南
从这里开始
- 先读完整个 Issue,再读项目的贡献指南。
- 在 Issue 下留言说明你要接手 —— 这能避免两个人做同样的事。
- Fork 仓库,在一个分支上完成修改。
- 提交 Pull Request,并在描述里引用这个 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