Pytorch libtorch（Visual Studio 2015） 部署经历（一）

一、下载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文件夹再次运行即可。