悲しいかな、nvcc --version
で取得できるバージョンはCUDAのみでこれではcuDNNのバージョンは取得できない
方法1 cudnn.hを見る
# cat /usr/local/cuda/include/cudnn.h | grep CUDNN_MAJOR -A 2
以下はcuDNN 6.0の場合(CUDNN_MAJOR=6 CUDNN_MINOR=0なので)
# cat /usr/local/cuda/include/cudnn.h | grep CUDNN_MAJOR -A 2 #define CUDNN_MAJOR 6 #define CUDNN_MINOR 0 #define CUDNN_PATCHLEVEL 21 -- #define CUDNN_VERSION (CUDNN_MAJOR * 1000 + CUDNN_MINOR * 100 + CUDNN_PATCHLEVEL) #include "driver_types.h"
方法2libcudnnを見る
ls -l /usr/local/cuda/lib64/libcudnn*
をしてシンボリックリンクがどのバージョンを向いているのかを見る方法もある
# ls -l /usr/local/cuda/lib64/libcudnn* lrwxrwxrwx 1 user users 13 4月 13 2017 /usr/local/cuda/lib64/libcudnn.so -> libcudnn.so.6 lrwxrwxrwx 1 user users 18 4月 13 2017 /usr/local/cuda/lib64/libcudnn.so.6 -> libcudnn.so.6.0.21 -rwxr-xr-x 1 user users 154322864 4月 13 2017 /usr/local/cuda/lib64/libcudnn.so.6.0.21 lrwxrwxrwx 1 user user 17 7月 26 2017 /usr/local/cuda/lib64/libcudnn.so.7 -> libcudnn.so.7.0.1 -rwxrwxr-x 1 user user 215118928 7月 26 2017 /usr/local/cuda/lib64/libcudnn.so.7.0.1 -rw-r--r-- 1 user users 143843808 4月 13 2017 /usr/local/cuda/lib64/libcudnn_static.a
個人的には方法2のほうが確実な気がする。アップデートして、シンボリックリンクが意図していない方に向いてることは可能性としてあり得るし