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

举报
AI浩 发表于 2022/02/06 16:43:09 2022/02/06
【摘要】 ​ 最近想研究一下如何部署pytorch模型,看到别人说LibTorch不错,是最近流行的C++部署框架,我打算试试。首先,去pytorch官网上下载,如下图,上面的连接是release版本,下面是Debug版本,支持C++,但是Window不支持Java。点击连接下载下来。我选择的是release版本的。​将解压后的文件复制到C盘,如下图:​设置环境变量: 右键我的电脑->高级系统设置->...

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

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

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

设置环境变量:
 

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

.

然后新建C++工程文件

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

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


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


改完后的效果如下图:

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

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

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

在附件依赖项中增加:

c10.lib

mkldnn.lib

libprotobuf.lib

torch.lib

torch_cpu.lib

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

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

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


#include <torch/torch.h>

#include "torch/script.h"

#include <iostream>


int main()

{

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

    std::cout << output;


    return 0;

}

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

【版权声明】本文为华为云社区用户原创内容,转载时必须标注文章的来源(华为云社区)、文章链接、文章作者等基本信息, 否则作者和本社区有权追究责任。如果您发现本社区中有涉嫌抄袭的内容,欢迎发送邮件进行举报,并提供相关证据,一经查实,本社区将立刻删除涉嫌侵权内容,举报邮箱: cloudbbs@huaweicloud.com
  • 点赞
  • 收藏
  • 关注作者

评论(0

0/1000
抱歉,系统识别当前为高风险访问,暂不支持该操作

全部回复

上滑加载中

设置昵称

在此一键设置昵称,即可参与社区互动!

*长度不超过10个汉字或20个英文字符,设置后3个月内不可修改。

*长度不超过10个汉字或20个英文字符,设置后3个月内不可修改。