下载libtorch
cd/root/Data/libs/cpp
wget https://download.pytorch.org/libtorch/nightly/cpu/libtorch-shared-with-deps-latest.zip
unzip libtorch-shared-with-deps-latest.zip
创建流程
初始化
cd /root/Projects/myProjects/cppProjects
mkdir torchTest && cd torchTest
编写源码example-cpp.cpp
#include <torch/torch.h>
#include <iostream>
int main() {
torch::Tensor tensor = torch::rand({2, 3});
std::cout << tensor << std::endl;
}
编写CMakeLists.txt
cmake_minimum_required(VERSION 3.0 FATAL_ERROR)
project(torchTest)
find_package(Torch REQUIRED)
set(CMAKE_CXX_FLAGS "${CMAKE_CXX_FLAGS} ${TORCH_CXX_FLAGS}")
add_executable(torchTest example-app.cpp)
target_link_libraries(torchTest "${TORCH_LIBRARIES}")
set_property(TARGET torchTest PROPERTY CXX_STANDARD 14)
编译项目
mkdir build && cd build
cmake -DCMAKE_PREFIX_PATH=/root/Data/libs/cpp/libtorch ..
cmake --build . --config Release
如图: