libtorch在windows上的使用及采坑

  • Post author:
  • Post category:其他


1.环境搭建

2.将pytorch模型导出成pt

3.使用libtorch c++ api调用加载模型

4.errorlist#1:libtorch在windows上部署时出现这个报错,模型无法加载:

libtorch在windows上部署时出现这个报错,模型无法加载;解决方案:

在C++->属性->链接器->命令行添加:

/INCLUDE:”?warp_size@cuda@at@@YAHXZ”

/INCLUDE:”?ignore_this_library_placeholder@@YAHXZ”

首先得确保你的libtorch和torch版本一致,不然都是白搭

在使用debug下使用CPU能进行模型测试,但使用GPU出现C10::error;

在img.to(device)->img=img.to(device)

5.errorlist#2: Relsease或者Debug只有一个能够跑通:

解决方案:debug和release需要分开配置,库不能够共用

6.参考链接

libtorch使用教程,知乎上总结


libtorch (pytorch c++) 教程(一) – 知乎



版权声明:本文为whunamikey原创文章,遵循 CC 4.0 BY-SA 版权协议,转载请附上原文出处链接和本声明。