Расширения 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?