C++调用libtorch1.12遇到的几个问题

先说一下我对原理并不懂,只是记录一下自己遇到的网上没查到的解决方法,仅供参考。
系统为windows

1、无法打开文件caffe2_detectron_ops_gpu.lib

这是因为我下载的libtorch1.12.1的库里面没有这些.lib,直接删掉就好,包括:
caffe2_module_test_dynamic.lib
mkldnn.lib

2、加载模型报错c10::Error

使用 torch::jit::script::Module module = torch::jit::load(“./*.pt”) 时候报错 Microsoft C++ 异常: c10::Error,位于内存位置 0x000000C6DF4F8EF0 处。

我是因为C++编译为DEBUG版本,但是下载的libtorch是release版本,所以将libtorch改为debug版本,然后重新配置再编译各种链接库就好了

当然网上大多数都是libtorch和pytorch的版本对齐问题,CUDA或CPU对齐问题。

3、使用GPU报错c10::NotImplementedError

这出现在to(device)时候,首先确认torch::cuda::is_available() 返回1,即可以使用。
一般网上都是说添加 /INCLUDE:"?searchsorted_cuda@native@at@@YA?AVTensor@2@AEBV32@0_N1@Z" 到属性->链接器->命令行中。但我添加后始终无法编译。
这篇github 提到最新版本使用 /INCLUDE:?warp_size@cuda@at@@YAHXZ /INCLUDE:?_torch_cuda_cu_linker_symbol_op_cuda@native@at@@YA?AVTensor@2@AEBV32@@Z
问题解决。

猜你喜欢

转载自blog.csdn.net/longshaonihaoa/article/details/127409819