质量为本、客户为根、勇于拼搏、务实创新
介绍
CUDNN是一个针对CUDA编程的深度神经网络加速库,由NVIDIA公司开发,可以大幅提升深度神经网络训练和推断的速度。如果你使用CUDA进行深度学习开发,那末CUDNN或许是你不可或缺的工具之一。本文将介绍怎么查看Linux系统下的CUDNN版本。
查看方法
在Linux系统下,可以通过终端上运行以下命令来查看CUDNN的版本号:
$cat/usr/local/cuda/include/cudnn.h|grepCUDNN_MAJOR-A2履行以上代码后应当会看到类似下面的输出:
defineCUDNN_MAJOR8defineCUDNN_MINOR2defineCUDNN_PATCHLEVEL0--defineCUDNN_VERSION(CUDNN_MAJOR1000+CUDNN_MINOR100+CUDNN_PATCHLEVEL)其中,CUDNN_MAJOR表示主版本号,CUDNN_MINOR表示次版本号,CUDNN_PATCHLEVEL表示修订版本号。通过这三个数字便可准确肯定当前系统上安装的CUDNN版本。在本例中,输出结果表明当前系统上安装的CUDNN版本号为8.2.0。
注意事项
在使用CUDNN时,我们也需要注意以下几个问题:
通过上述方法,你就可以轻松地查看当前Linux系统下的CUDNN版本,保证你的深度学习程序运行在正确的环境中。
桂,哥,网,络www.GuIgege.cn
1对1专业客服
24小时服务支持
365天无间断服务
5分钟快速响应TOP