search

lilanxiao / Rotated_IoU

Differentiable IoU of rotated bounding boxes using Pytorch
MIT License
416 stars 64 forks source link

CUDA 11 #14

Closed filaPro closed 3 years ago

filaPro commented 3 years ago

Hi @lilanxiao , Looks like this code can't be compiled with cuda11. Can you please have a look at it?

lilanxiao commented 3 years ago

Hi @filaPro ,

I've only tested my code with Pytorch 1.5 and CUDA-Toolkit-10.2. But the problem is probably not the CUDA version, since I only use the most basic CUDA APIs. According to the home page of Pytorch, it supports CUDA 11 since 1.7.1 version. I assume that you compiled the code with a version newer than 1.5.

I guess the Pytorch team has probably changed some APIs of its C++ backend in the 1.6 or 1.7, so that my code cannot be compiled anymore. I've seen a similar issue when I tried to compile some legacy code written at the time of Pytorch 1.0 with Pytorch 1.5. I saw tons of errors from the C++ backend of Pytorch. Simply updating the legacy code with newer APIs solved the issue. CUDA version didn't really matter.

I'm sorry that I don't have enough time to add support for a newer Pytorch, since I'm struggling with my own deadlines. I suggest you check the C++ APIs of Pytorch. I hope it would help.