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数据类型,但是我用的时候发现这个数据类型会报错,于是我就直接删除掉了这个数据类型。