公司服務器使用了GeForce RTX 3090顯卡,該顯卡目前只能安裝cuda11.0版本,所以需要安裝pytorch1.7,在編譯DCNv2時出現“error: identifier “THCState_getCurrentStream“ is undefined”錯誤。
其他人的解決方案是降低pytorch版本,但低版本pytorch無法兼容cuda11.0。
最終找到解決方案:
Github下載:https://github.com/MatthewHowe/DCNv2
下載該工程,將DCN目錄下的dcn_v2.py文件移動到上層,和make.sh文件放在同一級目錄,然后執行:./make.sh

編譯成功!
