focal loss 包导入问题

[marxmarma]

楼主您好：

首次运行时发现sigmoid_focal_loss_cuda.cpython-37m-x86_64-linux-gnu.so 包导入出现问题。

具体报错：
File "/home/admin/pytorch_solov2-master/modules/sigmoid_focal_loss/sigmoid_focal_loss.py", line 5, in
from . import sigmoid_focal_loss_cuda

ImportError: /home/admin/pytorch_solov2-master/modules/sigmoid_focal_loss/sigmoid_focal_loss_cuda.cpython-37m-x86_64-
linux-gnu.so:

undefined symbol: _ZN2at5emptyEN3c108ArrayRefIlEERKNS0_13TensorOptionsENS0_8optionalINS0_12MemoryFormatEEE

[ABlueLight]

尝试使用torch==1.5.1 torchvision==0.6.1试试呢

[guomaotianjie]

@marxmarma you should first run setup.py then generate.so according you env