CUDA, cuDNNのバージョン確認
CUDA のバージョン
テキストから取得する
terminal
$ cat /usr/local/cuda/version.txt
CUDA Version 10.0.130
バージョンコマンドで取得する
terminal
$ nvcc -V
nvcc: NVIDIA (R) Cuda compiler driver
Copyright (c) 2005-2015 NVIDIA Corporation
Built on Tue_Aug_11_14:27:32_CDT_2015
Cuda compilation tools, release 7.5, V7.5.17
この方法もよく紹介されているのですが,うまく動いていない可能性があります.
上の方法と下の方法は同じ環境下で実行していますが,見ての通りバージョンが異なっているからです.
このコマンドで取得しているのは,PATH が通っている /usr/bin/nvcc
のようです.
terminal
$ /usr/bin/nvcc -V
nvcc: NVIDIA (R) Cuda compiler driver
Copyright (c) 2005-2015 NVIDIA Corporation
Built on Tue_Aug_11_14:27:32_CDT_2015
Cuda compilation tools, release 7.5, V7.5.17
コマンドを直接呼ぶと正しく表示されます.
terminal
$ /usr/local/cuda/bin/nvcc --version
nvcc: NVIDIA (R) Cuda compiler driver
Copyright (c) 2005-2018 NVIDIA Corporation
Built on Sat_Aug_25_21:08:01_CDT_2018
Cuda compilation tools, release 10.0, V10.0.130
ちなみに,nvidia-smi
でも CUDA のバージョンらしきものは表示されますが,ドライバが対応している CUDA のバージョンを表示しているに過ぎないようです.
terminal
$ nvidia-smi
Sat May 23 17:20:28 2020
+-----------------------------------------------------------------------------+
| NVIDIA-SMI 440.33.01 Driver Version: 440.33.01 CUDA Version: 10.2 |
|-------------------------------+----------------------+----------------------+
| GPU Name Persistence-M| Bus-Id Disp.A | Volatile Uncorr. ECC |
| Fan Temp Perf Pwr:Usage/Cap| Memory-Usage | GPU-Util Compute M. |
|===============================+======================+======================|
| 0 Tesla V100-PCIE... Off | 00000000:1A:00.0 Off | 0 |
| N/A 43C P0 38W / 250W | 0MiB / 16160MiB | 0% Default |
+-------------------------------+----------------------+----------------------+
| 1 Tesla V100-PCIE... Off | 00000000:1B:00.0 Off | 0 |
| N/A 44C P0 37W / 250W | 0MiB / 16160MiB | 0% Default |
+-------------------------------+----------------------+----------------------+
| 2 Tesla V100-PCIE... Off | 00000000:3E:00.0 Off | 0 |
| N/A 44C P0 37W / 250W | 0MiB / 16160MiB | 3% Default |
+-------------------------------+----------------------+----------------------+
+-----------------------------------------------------------------------------+
| Processes: GPU Memory |
| GPU PID Type Process name Usage |
|=============================================================================|
| No running processes found |
+-----------------------------------------------------------------------------+
参考:How to get the cuda version? | Stack Overflow
cuDNN のバージョン
terminal
$ cat /usr/include/cudnn.h | grep CUDNN_MAJOR -A 2
#define CUDNN_MAJOR 7
#define CUDNN_MINOR 6
#define CUDNN_PATCHLEVEL 5
--
#define CUDNN_VERSION (CUDNN_MAJOR * 1000 + CUDNN_MINOR * 100 + CUDNN_PATCHLEVEL)
#include "driver_types.h"
下のコマンドでも取得できるようです.(弊環境ではファイルがなかった)
terminal
$ cat /usr/local/cuda/include/cudnn.h | grep CUDNN_MAJOR -A 2