99网
您的当前位置:首页ubuntu如何查看已安装cudnn版本

ubuntu如何查看已安装cudnn版本

来源:99网

由于在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

因篇幅问题不能全部显示,请点此查看更多更全内容