Ubuntu18.04+RTX3090+pytorch1.7 安裝Yolact無法編譯DCNv2解決方法


公司服務器使用了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

 

 編譯成功!


免責聲明!

本站轉載的文章為個人學習借鑒使用,本站對版權不負任何法律責任。如果侵犯了您的隱私權益,請聯系本站郵箱yoyou2525@163.com刪除。



 
粵ICP備18138465號   © 2018-2025 CODEPRJ.COM