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

评论 1
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值