Windows上的PyTorch扩展-为什么未解析的符号(THFloatTensor等)?

时间:2019-06-28 15:00:13

标签: windows pytorch

PyTorch扩展似乎是实验性的。 这里有解决方法: https://pytorch.org/docs/stable/notes/windows.html#extension

我转换为cpp并按如下所述添加了库:

from torch.utils.ffi import create_extension

ffi = create_extension(
  '_ext.nms',
  headers=headers,
  sources=sources,
  define_macros=defines,
  relative_to=__file__,
  with_cuda=with_cuda,
  extra_objects=extra_objects,
  libraries=["cudart","_C"]
)

它可以编译,但是,我仍然从链接器中收到很多未解析的外部符号:

nms.obj : error LNK2001: unresolved external symbol __imp_THFloatTensor_data
nms.obj : error LNK2001: unresolved external symbol __imp_THByteTensor_fill
nms.obj : error LNK2001: unresolved external symbol __imp_THExpm1
[...]

_C.lib似乎不包含TH ...符号(已通过dumpbin _C.lib / ALL检查),并且我找不到任何libTH.lib或类似链接。

在哪里可以找到合适的库? 在Windows上有机会这样做吗?

0 个答案:

没有答案