关于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