Libtorch-1.7.1+cu101+opencv

Libtorch安装

1、解压Libtorch

• 建立cmake目录

mkdir build

按如下构建目录

• 书写CMakeLists.txt

cmake_minimum_required(VERSION 3.0 FATAL_ERROR)
project(custom_ops)
set(CMAKE_PREFIX_PATH "/mnt/data/lzhou/code/libtorch")
find_package(Torch REQUIRED)

add_executable(example-app example-app.cpp)
target_link_libraries(example-app "${TORCH_LIBRARIES}")
set_property(TARGET example-app PROPERTY CXX_STANDARD 14)

• 然后cmake

cd build
cmake -DCMAKE_PREFIX_PATH=/mnt/data/lzhou/code/libtorch/ ..
cmake --build . --config Release

• 修改CMakeCache.txt

在这个过程中，会有cudnn的要求，安装cudnn,但是安装后环境变量设置之后还是会有问题，那就需要修改build下面的CMakeCache.txt，将NOT-FOUND进行修改，其他安装包如果NOT-FOUND也需要修改CMakeCache.txt。

对CMakeCache.txt的CUDNN_LIBRARY_PATH，CUDNN_INCLUDE_PATH手动设置

CUDNN_INCLUDE_DIR:PATH=/mnt/data/lzhou/caffe_rely/cuda/include

//Path to a file.
CUDNN_INCLUDE_PATH:PATH=/mnt/data/lzhou/caffe_rely/cuda/include

//Path to the cudnn library file (e.g., libcudnn.so)
CUDNN_LIBRARY:PATH=/mnt/data/lzhou/caffe_rely/cuda/lib64

//Path to a library.
CUDNN_LIBRARY_PATH:FILEPATH=/mnt/data/lzhou/caffe_rely/cuda/lib64/libcudnn.so

• 编译

make -j

• 执行

 ./example-app

Opencv-3.4.9安装

要注意的是BUILD_TIFF要打开，opencv N个人安装会有N种奇怪的bug，只能 根据bug调整cmake选项。

unzip opencv-3.4.9.zip

cd opencvxxxx

mkdir build && cd build

cmake -D CMAKE_BUILD_TYPE=RELEASE -D cmake -D CMAKE_BUILD_TYPE=RELEASE -D CMAKE_INSTALL_PREFIX=/home/lzhou/local/opencv-3.4.9 -D WITH_TBB=OFF -D BUILD_opencv_gpu=OFF -D CUDA_GENERATION=Kepler -DBUILD_opencv_cudacodec=OFF -D BUILD_TIFF=ON ..

make -j4

make install

libtorch和opencv联合编译

只需要修改CMakeList.txt，包含多个cpp和h文件。

cmake_minimum_required(VERSION 3.0 FATAL_ERROR)
project(custom_ops)
set(CMAKE_PREFIX_PATH "/home/lzhou/pytorch-file2/libtorch-example/libtorch")
find_package(Torch REQUIRED)
find_package(OpenCV REQUIRED )
add_executable(example-app example-app.cpp my_inference.cpp my_inference.h)
target_link_libraries(example-app "${TORCH_LIBRARIES}")
target_link_libraries(example-app "${OpenCV_LIBS}")
set_property(TARGET example-app PROPERTY CXX_STANDARD 14)