libtorch/share/cmake/Caffe2/public/cuda.cmake:151

使用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)