libtorch/share/cmake/Caffe2/public/cuda.cmake:151
在cuda.cmake 147行处（if(CAFFE2_USE_CUDNN)上）添加 set(CUDNN_INCLUDE_PATH /usr/local/cuda/include)cuda 一般是安装在/usr/local/cuda 但是caffe2中的cmake默认路径是/usr/lib/cuda。使用libtorch cmake 出现该错误。出现这个错误的前提是你已经安装了cudnn。所以自然
·
使用libtorch cmake 出现该错误
file failed to open for reading (No such file or directory): /usr/lib/cuda/include/cudnn.h
出现这个错误的前提是你已经安装了cudnn
原因：
cuda 一般是安装在/usr/local/cuda 但是caffe2中的cmake默认路径是/usr/lib/cuda
所以自然会出现这个错误
解决办法（修改cuda.cmake 指定你自己的cuda路径）：
在cuda.cmake 147行处（if(CAFFE2_USE_CUDNN)上）添加 set(CUDNN_INCLUDE_PATH /usr/local/cuda/include)
有“AI”的1024 = 2048，欢迎大家加入2048 AI社区
更多推荐
0
0- 0
已为社区贡献1条内容
所有评论(0)