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

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

然后打开项目属性表,首先来配置包含目录
VS2017 + Libtorch + CPU环境配置及模型读取
库目录:
VS2017 + Libtorch + CPU环境配置及模型读取

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

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