Libtorch + vs 2019安装及配置

Libtorch是c++接口的Pytorch,提供了与python版本基本一致的接口,使用很方便,而且更适合实际部署。这里介绍Libtorch的在VS2019下的部署和简单测试使用。

1、从官网下载

https://pytorch.org/get-started/locally/

目前可以下载到Libtorch1.5版本,如果电脑支持Gpu,且安装了Cuda,可以下载Gpu版本,否则下载cpu版本。

 

下载解压之后如下

 

2、在VS2019中的配置

需要配置.头文件h 库文件,lib, 和dll文件

配置头文件

 

配置依赖库的位置

 

添加依赖库的文件名

 

设置dll文件的环境变量,

PATH=D:\software\gnsoftware\pyTorch\libtorch-win-shared-with-deps-debug-1.5.0\libtorch\lib;%PATH%

(之前我一般是把那一堆dll文件拷贝到生成的exe文件所在目录,导致这些dll文件会复制很多份,占据很大空间,这里通过设置依赖于项目的环境变量,少了大量拷贝,还不影响其他变量)

 

4、简单的测试例子

#include <iostream>
#include <torch/torch.h>
 
int main()
{
    
    torch::Tensor tensor = torch::rand({ 5,3 });
    std::cout << tensor << std::endl;
 
    return EXIT_SUCCESS;
}


运行结果如下,说明配置没问题,可以开始后面的Libtorch学习之旅。

 

  • 0
    点赞
  • 0
    收藏
    觉得还不错? 一键收藏
  • 0
    评论

“相关推荐”对你有帮助么?

  • 非常没帮助
  • 没帮助
  • 一般
  • 有帮助
  • 非常有帮助
提交
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

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

抵扣说明:

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

余额充值