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上有机会这样做吗?