How can i compile tensorflow c++ API on jetson xavier nx?

tanpopo.chien · March 25, 2022, 1:19am

I try compile tensorflow 1.15.5 from source on nx, but get the error below:

ERROR: /home/mrobot/workspace/tensorflow-1.15.5/tensorflow/stream_executor/cuda/BUILD:322:1: C++ compilation of rule ‘//tensorflow/stream_executor/cuda:cudnn_stub’ failed (Exit 1)
In file included from tensorflow/stream_executor/cuda/cudnn_stub.cc:57:0:
./tensorflow/stream_executor/cuda/cudnn_7_1.inc:626:33: error: ‘cudnnConvolutionFwdPreference_t’ has not been declared
cudnnConvolutionFwdPreference_t preference,
^~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~

any one can help?

AastaLLL · March 25, 2022, 2:40am

Hi,

It looks like a compatible issue.

Based on the error, the TensorFlow source tries to find cuDNN 7.1 API.
But if you are using JetPack 4.6, the cuDNN package is already 8.2.1.

Thanks.

tanpopo.chien · March 25, 2022, 3:16am

Hi,

thanks! yes, my J4T version is 4.6.

is downgrade cnDNN version to 7.1 can solve this problem?
but i think it may cause conflicts with other installed package, such lick TensorRT…

AastaLLL · March 31, 2022, 3:51am

Hi,

To downgrade the cuDNN package directly might cause some dependency issues.

Based on the below source, TensorFlow v1.15.x only supports cuDNN up to version 7.
It’s recommended to downgrade the whole JetPack software or use a compatible TensorFlow v2.x instead.

github.com

tensorflow/tensorflow/blob/v1.15.5/tensorflow/stream_executor/cuda/cudnn_stub.cc

/* Copyright 2015 The TensorFlow Authors. All Rights Reserved.

Licensed under the Apache License, Version 2.0 (the "License");
you may not use this file except in compliance with the License.
You may obtain a copy of the License at

    http://www.apache.org/licenses/LICENSE-2.0

Unless required by applicable law or agreed to in writing, software
distributed under the License is distributed on an "AS IS" BASIS,
WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
See the License for the specific language governing permissions and
limitations under the License.
==============================================================================*/
#include "third_party/gpus/cudnn/cudnn.h"
#include "tensorflow/stream_executor/lib/env.h"
#include "tensorflow/stream_executor/platform/dso_loader.h"

// Implements the cuDNN API by forwarding to cuDNN loaded from the DSO.

This file has been truncated. show original

Thanks.

system · April 20, 2022, 7:34am

This topic was automatically closed 14 days after the last reply. New replies are no longer allowed.