cmake链接cudnn

linux使用cuda时,cuda一般是安装的,cudnn是解压后放到指定目录的。如果将cudnn放到cuda目录下就没有下面的工作了。

在CMakeLists.txt中没有指定cudnn包含路径和库路径时使用下面的命令链接cudnn:
 

cmake .. -DCUDNN_INCLUDE_DIR=../../cuda/include -DCUDNN_LIBRARY=../../cuda/lib64

此时的cudnn路径在CMakeLists.txt所在路径的上上级目录下的cuda文件夹下。

如果在CMakeLists.txt中写上这些路径,编译时只需要:

cmake ..

CMakeLists.txt中写上这些路径:

set(CUDNN_INCLUDE_DIR=../../cuda/include)
set(CUDNN_LIBRARY=../../cuda/lib64)

猜你喜欢

转载自blog.csdn.net/juluwangriyue/article/details/114018132