linux下使用python脚本查看CUDA和CUDNN版本

先说说用shell命令查看CUDA和CUDNN版本的方法：

查看CUDA版本的命令如下：

cat /usr/local/cuda/version.txt

查看CUDNN版本：

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

其实CUDA和CUDNN的版本信息写在对应的文件里，如果用上述命令找不到版本信息，那么就用

whereis cuda
whereis cudnn

来找到对应的文件，再替换上面命令中的文件路径。 

---

如果用Python脚本来查看CUDNN版本的话，就更为直观而且更为灵活了，脚本如下：

import os
os.system('cat /usr/local/cuda/version.txt')
#os.system('whereis cudnn')
try:
    head_file = open('/usr/local/cuda/include/cudnn.h')
except:
    head_file = open('/usr/include/cudnn.h')
lines = head_file.readlines()
for line in lines:
    line = line.strip()
    if line.startswith('#define CUDNN_MAJOR'):
        line = line.split('#define CUDNN_MAJOR')
        n1 = int(line[1])
        continue
    if line.startswith('#define CUDNN_MINOR'):
        line = line.split('#define CUDNN_MINOR')
        n2 = int(line[1])
        continue
    if line.startswith('#define CUDNN_PATCHLEVEL'):
        line = line.split('#define CUDNN_PATCHLEVEL')
        n3 = int(line[1])
        break
print('CUDNN Version ', str(n1)+'.'+str(n2)+'.'+str(n3))

把上述代码复制到一个.py文件中，比如命名为cudnn_version.py

只要python cudnn_version.py就可以查看到直观信息了，输出信息如下：