Caffe installation TX2 | Check Failed: CUBLAS_STATUS_SUCCESS (1 vs. 0)

Hello,

I’m getting the following error during “make runtest” on the Caffe install. “make all” and “make test” both ran to completion with no errors.

RMSPropSolverTest/3.TestRMSPropLeastSquaresUpdateWithEverythingShare
F0907 14:24:32.073608 32570 math_functions.cu:42] Check failed: status == CUBLAS_STATUS_SUCCESS (1 vs. 0) CUBLAS_STATUS_NOT_INITIALIZED
*** Check failure stack trace: ***

System:
Nvidia TX2 with jetpack 32
CUDA 9, CUDNN 7.1

Makefile.config modification (from MakeFile.config.example)

USE_CUDNN := 1 #enables build with CUDNN
OPENCV_VERSION := 3 #sets opencv build version
CUDA_ARCH := -gencode arch=compute_62,code=sm_62

Hi,

Suppose you want to use caffe for inferencing.

There are some caffe test items cannot be executed on the Jetson platform.
But it should be okay if you only use it for inferencing.

You can step up caffe with these steps:
https://devtalk.nvidia.com/default/topic/1021931/jetson-tx2/nvcaffe-support-on-tx2/

And just ignore the failed test items.
Thanks.