error: identifier 「THCState_getCurrentStream「 is undefined

問題 error: identifier 「THCState_getCurrentStream」 is undefined … AT_ASSERTM(trans.type().is_cuda(), 「trans must be a CUDA tensor」); 在對DCNv2模型進行編譯的時候出現了報錯,記錄一下解決辦法。 解決 1、Google: 似乎是 pytorch 版本的問題!好不容易纔安
相關文章
相關標籤/搜索