install cuDNN in TX2 , err : CUDNN_STATUS_BAD_PARAM

Robotics & Edge Computing Jetson & Embedded Systems Jetson TX2
1

Hi all:

I install cuDNN in TX2 , compiling caffe passed , but when I run it ,

system prompt : CUDNN_STATUS_BAD_PARAM(3 vs 0…, and then be interupted , who can help me ??

Ubuntu 16.04
CUDA 8.0
cudnn 5.0.10

B/R

2

Hi,

Have you corrected the compiler option for TX2?

Here is an example for your reference:

Thanks

3

Hi AastaLLL,

thanks for your response , but the installCaffe.sh is very sample , there is no any detail about CuDNN …

BTW, I do not run the step : $ sudo ./jetson_clocks.sh

if it is necessary??

B/R

4

My detail err be listed as following :

CUDNN_STATUS_SUCCESS (3 vs. 0) CUDNN_STATUS_BAD_PARAM

the compiler option of CUDA be listed as following :

USE_CUDNN := 1

CUDA_ARCH :=-gencode arch=compute_35,code=sm_35
-gencode arch=compute_50,code=sm_50
-gencode arch=compute_50,code=compute_50

5

Hi,

TX2 is in 6.2 GPU architecture: CUDA GPUs - Compute Capability | NVIDIA Developer
Please add it into your configure file:

-gencode arch=compute_62,code=sm_62 \

Thanks.