cmake链接cudnn
linux使用cuda时,cuda一般是安装的,cudnn是解压后放到指定目录的。如果将cudnn放到cuda目录下就没有下面的工作了。在CMakeLists.txt中没有指定cudnn包含路径和库路径时使用下面的命令链接cudnn:cmake .. -DCUDNN_INCLUDE_DIR=../../cuda/include -DCUDNN_LIBRARY=../../cuda/lib64此时的
·
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)
更多推荐
已为社区贡献3条内容
所有评论(0)