pandazx's blog

Hadoop, データ分析など雑多な技術ブログ

cuDNNのバージョン確認

cudaのインストールディレクトリが /usr/local/cuda と仮定する。その場合、以下に書いてある。

  • /usr/local/cuda/include/cudnn.h のCUDNN_MAJOR, CUDNN_MINOR, CUDNN_PATCHLEVEL
  • /usr/local/cuda/lib64/libcudnn.so.x.x.x

x.x.xがバージョン

以下の方法もある。

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

or

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

このコマンドは下記ページからの引用。

medium.com

ちなみに、CUDAのバージョン確認は nvcc -V

GPUとの疎通確認は nvidia-smi