一、下载libtorch
二、在任意位置新建1个文件夹,并在文件夹新建2个文件
- example-app.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(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文件夹新建一个build文件夹,
在cmake中设置source code与build路径
点击configure,选择VS2015 X64版本,这时会报错,由于libtorch的地址没有传入,此时保持Cmake GUI开启,在build文件夹中运行如下命令行(注意:后面有两个点)
cmake -DCMAKE_PREFIX_PATH=libtorch绝对地址 ..
没有错误的话,就说明成功了。
– Found torch: F:/libtorch/lib/torch.lib
– Configuring done
– Generating done
– Build files have been written to: C:/Desktop/pytorch_test/build
直接运行生成的vs解决方案,就可以正常运行了。
若缺少DLL文件,将libtorch/lib文件夹中相关的文件拷贝到Debug/Release文件夹再次运行即可。