VS2015调用pytorch训练好的模型

最新推荐文章于 2024-08-05 15:53:03 发布

若曦爹

最新推荐文章于 2024-08-05 15:53:03 发布

阅读量837

点赞数

本文链接：https://blog.csdn.net/gbz3300255/article/details/115081107

版权

本文详细介绍了如何在VS2015中调用PyTorch训练的模型。首先，确保PyTorch和libtorch版本匹配，下载正确版本的libtorch。接着，参照教程配置VS2015环境。然后，将PyTorch模型转换为Torch Script，并在C++代码中使用`torch::jit::load`加载模型。在过程中可能会遇到错误，如C4146和C4996，解决方法分别是对无符号类型进行修正和添加预处理器定义 `_CRT_SECURE_NO_WARNINGS`。

摘要由CSDN通过智能技术生成

1.最重要的一点，pytorch最好与libtorch版本相同。例如我的pyotrch版本是1.4.0的。在这篇文章

https://blog.csdn.net/qq_37569355/article/details/104246813

里找到

https://download.pytorch.org/libtorch/cpu/libtorch-win-shared-with-deps-1.0.0.zip这样的字样。

然后我把1.0.0给改成1.4.0

https://download.pytorch.org/libtorch/cpu/libtorch-win-shared-with-deps-1.4.0.zip

就能下载到我的纯cpu的1.4.0的libtorch的东西了啊。

如果下载不对会出如下错误的

å¨è¿éæå¥å¾çæè¿°

2.下载完libtorch就去搜个 libtorch与vs2015的安装方法吧

参考这个就好

https://blog.csdn.net/weixin_38376691/article/details/109263676

3.将自己pytorch训练的模型转成Torch Script

注意：我是将网络也写在里面了。我用的VGGnet 主要为了方便。不把网络写里面会报错的哦.这个是pytorch代码

import torch
import torchvision
import torch.nn as nn


###################################################### VGGNET ##################################