ubuntu查看cuda版本和cudnn版本

1、查看cuda版本

cat /usr/local/cuda/version.txt

2、查看cudnn版本

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

根据输出判断：eg：

2. #define CUDNN_MAJOR 5
3. #define CUDNN_MINOR 1
4. #define CUDNN_PATCHLEVEL 5
5. --
6. #define CUDNN_VERSION (CUDNN_MAJOR * 1000 + CUDNN_MINOR * 100 + CUDNN_PATCHLEVEL)

CuDNN版本为 5.1.5

我的输出为：说明版本为6.0.21

2. #define CUDNN_MAJOR 6
3. #define CUDNN_MINOR 0
4. #define CUDNN_PATCHLEVEL 21
5. --
6. #define CUDNN_VERSION (CUDNN_MAJOR * 1000 + CUDNN_MINOR * 100 + CUDNN_PATCHLEVEL)
7.  
8. #include "driver_types.h