tensorflow / java

Java bindings for TensorFlow
Apache License 2.0
813 stars 200 forks source link

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

Open Jefftu1024 opened 1 year ago

Jefftu1024 commented 1 year ago

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

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

Craigacp commented 1 year ago

When you ran it after upgrading cuDNN it ran in Python? TF-Java doesn't have a python VM running, so what error message do you get out of TF-Java after cuDNN has been upgraded?

Jefftu1024 commented 1 year ago

I've got the same error like python. I forget print the screen so replace it with python. Now I reproducing it below. 图片 图片