- pytorch C++接口调用,最简单示例:
pytroch学习(二十一)—C++(libTorch)调用pytroch预训练模型
2个教程类型,步骤:下载pytorch 的C++ 接口代码,解压后即可使用,不用安装。
包括CMakeLists.txt的设置:
cmake_minimum_required(VERSION 3.0 FATAL_ERROR)
project(example-app)
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 11)
example-app.cpp:
#include <torch/torch.h>
#include <iostream>
int main() {
torch::Tensor tensor = torch::rand({2, 3});
std::cout << tensor << std::endl;
}
编译及运行:
cmake .
make
./example-app
- 一个错误:cmake .的时候找不到
TorchConfig.cmake
和torch-config.cmake
文件, 解决是在CMakeLists.txt中设置,
set( CMAKE_PREFIX_PATH "/****/libtorch-cxx11-abi-shared-with-deps-1.4.0/libtorch/share/cmake/Torch")#***部分写自己的路。来自关于windows10用c++部署libtorch过程中遇到的一些问题
- 具体C++中代码相关,包括opencv的调用
2个参考文章:
Libtorch踩坑实录:non-scalar type, '->' has non-pointer type,opencv,Expected Tensor but got Tuple