ubuntu檢視cuda/cudnn版本

ubuntu檢視cuda版本的指令：

nvcc -V
           

檢視cudnn版本的指令（老版本）：

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

檢視cudnn版本的指令（新版本）：

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

注意：可能cudnn不是裝在/usr目錄下的，需要根據自己的cudnn安裝路徑對cudnn.h或cudnn_version.h的路徑進行調整