Media-Smart / vedadet

A single stage object detection toolbox based on PyTorch
Apache License 2.0
498 stars 128 forks source link

Facing issue when running "pip install -v -e ." on windows machine #56

Closed kiranbab closed 5 months ago

kiranbab commented 3 years ago

Facing issue when running "pip install -v -e ." on windows machine config{ windows:10 cuda 10.2 python:3.8}

image

darjazafirovic commented 2 years ago

As explained here https://forums.developer.nvidia.com/t/strange-link-error-seen-by-multiple-people-while-building-pytorch-cpp-cuda-extensions/145261/2 the issue happens because long resolves to 32-bit int on Windows as opposed to 64-bit long long on Linux. Replace the two occurrences of long in sigmoid_focal_loss_cuda.cu with int64_t and it will work fine.