[ubuntu]ubuntu如何查看已安装cudnn版本

cudnn由于在8.0版本前后版本存储头文件不一样,所以需要分情况查询,如果你是8.0版本以前可以输入:

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

如果你是在8.0以后可以用下面命令:

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

下面我创造一个通用查询方法,支持所有版本cudnn版本查询:

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

猜你喜欢

转载自blog.csdn.net/FL1623863129/article/details/131493930