I apply the Caffe installation on the Jetson ,an error came out

282464532 · September 25, 2020, 12:28pm

Jetpack Version: 4.4-b186
CUDA 10.2
cuDNN 8.0.0

Follow post:caffe for jetson nano?, I installed caffe with USE_CUDNN := 1, an error came out.
./include/caffe/util/cudnn.hpp:15:28: note: in definition of macro ‘CUDNN_CHECK’
cudnnStatus_t status = condition;
^~~~~~~~~
src/caffe/layers/cudnn_conv_layer.cpp:149:59: error: ‘cudnnGetConvolutionBackwardFilterAlgorithm’ was not declared in this scope.

AastaLLL · September 28, 2020, 4:13am

Hi,

There are some API changes in cuDNN v8.0.
cudnnGetConvolutionBackwardFilterAlgorithm is removed and please use cudnnFindConvolutionBackwardFilterAlgorithm instead.

You can find a similar update here:

github.com

AastaNV/JEP/blob/master/script/topics/0001-fix-for-cudnn_v8.patch

From 9a3a2308a24f488e303e8389cf900c0cc53d8e71 Mon Sep 17 00:00:00 2001
From: Your Name <you@example.com>
Date: Fri, 10 Jul 2020 12:15:07 +0800
Subject: [PATCH] fix for cudnn_v8

---
 src/convolutional_layer.c | 64 +++++++++++++++++++++++++++++++++++++++
 1 file changed, 64 insertions(+)

diff --git a/src/convolutional_layer.c b/src/convolutional_layer.c
index 1fb58b0..9fed951 100644
--- a/src/convolutional_layer.c
+++ b/src/convolutional_layer.c
@@ -8,6 +8,8 @@
 #include <stdio.h>
 #include <time.h>
 
+#define PRINT_CUDNN_ALGO 0
+
 #ifdef AI2

This file has been truncated. show original

Thanks.

282464532 · October 3, 2020, 7:07am

It’s greate， Thank you very much.