LibTorch入门——Win10+VS2017配置LibTorch开发环境。

简介: LibTorch入门——Win10+VS2017配置LibTorch开发环境。

订阅专栏

最近想研究一下如何部署pytorch模型,看到别人说LibTorch不错,是最近流行的C++部署框架,我打算试试。


首先,去pytorch官网上下载,如下图,上面的连接是release版本,下面是Debug版本,支持C++,但是Window不支持Java。点击连接下载下来。我选择的是release版本的。

tt.png




将解压后的文件复制到C盘,如下图:

tt.png




设置环境变量:


右键我的电脑->高级系统设置->高级中的环境变量->点击系统变量中的Path->添加dll路径:


tt.png

.


然后新建C++工程文件

tt.png



新建完成后默认是x86的,需要改为X64的,因为libtorch是x64的,不支持32位的。我下载是release版本的,所以还要改为release模式。

tt.png




然后右键lib_torch项目,选择属性,配置libtorch环境。

tt.png





选择C\C++  ——》常规,将SDL检查改为否。添加附加包含目录。

tt.png

tt.png







改完后的效果如下图:

tt.png




将 C/C++ => 语言 => 符合模式,改为否。

tt.png




设置连接库,选择连接器——》常规,修改附加库目录

tt.png

tt.png






设置连接库,选择连接器——》输入,修改附加依赖项

tt.png




在附件依赖项中增加:


c10.lib


mkldnn.lib


libprotobuf.lib


torch.lib


torch_cpu.lib

tt.png




在 配置属性à调试à环境 里添加libtorch动态库的路径


PATH=C:\libtorch\lib;%PATH%

tt.png




到这里就配置完了,写入测试代码试一下。



#include <torch/torch.h>


#include "torch/script.h"


#include <iostream>



int main()


{


   torch::Tensor output = torch::randn({ 5,2 });


   std::cout << output;



   return 0;


}


出现下面的结果说明配置成功了。



tt.png

tt.png

目录
相关文章
|
1月前
|
开发工具 Windows
搭建 Qt 开发环境
搭建 Qt 开发环境
|
1月前
|
IDE Java 开发工具
【开发工具】Windows环境下 IDEA 开发工具安装&配置(无错完整)
【1月更文挑战第20天】【开发工具】Windows环境下 IDEA 开发工具安装&配置(无错完整)
|
9月前
c[win32]开发环境准备
c[win32]开发环境准备
79 0
|
11月前
|
Java
[BlocklyNukkit入门]#1配置开发环境
这里用PowerNukkit举例
75 0
|
IDE Java 编译器
Eclipse下STM32开发环境之安装配置(一)
Eclipse下STM32开发环境之安装配置(一)
405 1
Eclipse下STM32开发环境之安装配置(一)
|
Android开发 C语言 芯片
Eclipse下STM32开发环境之安装配置(二)
Eclipse下STM32开发环境之安装配置(二)
778 1
Eclipse下STM32开发环境之安装配置(二)
|
IDE 应用服务中间件 Go
go安装+Windows10+环境配置+idea配置+演示Demo【详细教程】
go安装+Windows10+环境配置+idea配置+演示Demo【详细教程】
744 0
|
IDE Unix Linux
Python入门开发: windows下搭建python开发环境(vscode)
Python入门开发: windows下搭建python开发环境(vscode)
210 0
Python入门开发: windows下搭建python开发环境(vscode)