关于cudnn版本的查看,大部分教程给的操作都是

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

但是新一点的cudnn都无法再通过这条指令查看版本号了

这是因为,新一些的cudnn版本信息都写在在cudnn_version.h而不是cudnn.h

所以复制h文件的时候,需要在

sudo cp cuda/include/cudnn.h /usr/local/cuda/include/
sudo cp cuda/lib64/libcudnn* /usr/local/cuda/lib64/
sudo chmod a+r /usr/local/cuda/include/cudnn.h
sudo chmod a+r /usr/local/cuda/lib64/libcudnn*

之后,再进行

sudo cp cuda/include/cudnn_version.h /usr/local/cuda/include/

查看版本号的指令也改为

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

 

Logo

欢迎来到由智源人工智能研究院发起的Triton中文社区,这里是一个汇聚了AI开发者、数据科学家、机器学习爱好者以及业界专家的活力平台。我们致力于成为业内领先的Triton技术交流与应用分享的殿堂,为推动人工智能技术的普及与深化应用贡献力量。

更多推荐