VS2017 + Libtorch + CPU环境配置及模型读取

VS2017 + Libtorch + CPU环境配置及模型读取
python端的还没弄,回头补上;
先说C++的问题
下载libtorch:https://pytorch.org/get-started/locally/
在这里插入图片描述
如图,因为我要配置的是CPU环境,所以下载的时候选择的是None。下载之后解压,放到指定位置,自己选择。
打开libtorch,能看到里面的文件夹
在这里插入图片描述
看见这些文件夹,很熟悉,和大多数的C++库差不多,那么久开始配置吧。
选择debug x64
在这里插入图片描述
添加项目属性表
在这里插入图片描述

然后打开项目属性表,首先来配置包含目录
在这里插入图片描述
库目录:
在这里插入图片描述

然后链接器里的附加库目录
在这里插入图片描述
附加依赖项
在这里插入图片描述
到这里还需要改两个属性:
C/C++的里面SDL检查改成否
在这里插入图片描述
然后再把C/C++里面的语言中的符合模式改成否
在这里插入图片描述
到这里,环境配置就算结束了,但是还有一步需要做,就是需要把附加依赖项的lib对应的dll复制到项目的cpp源文件的同目录下。到此为止,环境的配置就算结束了,然后就可以写代码了。

导入这几个头文件,写一段代码进行测试
在这里插入图片描述
在这里插入图片描述
在这里还要提一下,torch::jit::load的pt文件要通过python来输出,而且module的数据类型,很多地方都是std::shared_ptr数据类型,但是我用的时候发现这个数据类型会报错,于是我就直接删除掉了这个数据类型。

评论 3
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

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

抵扣说明:

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

余额充值