先说说用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就可以查看到直观信息了,输出信息如下: