检测Cudnn是否安装成功

Challenge

安装Cudnn后通过网上找到的方法（如下）没有任何输出，以为是没能安装成功。

cat /usr/local/cuda/include/cudnn.h | grep CUDNN_MAJOR -A2

或者

cat /usr/local/cuda/include/cudnn.h | grep CUDNN_MAJOR

或者

cat /usr/local/cuda-11.1/include/cudnn.h | grep CUDNN_MAJOR -A2

Solution

实际上，检测版本Cudnn是否安装成功的方法与老方法略有不同，头文件应该为cudnn_version.h而不是cudnn.h。以v8.2.1的版本为例：

cat /usr/local/cuda/include/cudnn_version.h | grep CUDNN_MAJOR -A 2