cudnn由于在8.0版本前后版本存储头文件不一样,所以需要分情况查询,如果你是8.0版本以前可以输入:
cat /usr/local/cuda/include/cudnn.h | grep CUDNN_MAJOR -A 2
如果你是在8.0以后可以用下面命令:
cat /usr/local/cuda/include/cudnn_version.h | grep CUDNN_MAJOR -A 2
下面我创造一个通用查询方法,支持所有版本cudnn版本查询:
cat /usr/local/cuda/include/cudnn*.h | grep CUDNN_MAJOR -A 2