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：

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

CuDNN版本為 5.1.5

我的輸出為：說明版本為6.0.21

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