install cuDNN in TX2 , err : CUDNN_STATUS_BAD_PARAM

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

Hi,

Have you corrected the compiler option for TX2?

Here is an example for your reference:

Thanks

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

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

Hi,

TX2 is in 6.2 GPU architecture: https://developer.nvidia.com/cuda-gpus
Please add it into your configure file:

-gencode arch=compute_62,code=sm_62 \

Thanks.