今天把之前在ubuntu下寫的pytorch的cuda擴(kuò)展放到win10下面編譯酌摇,記錄一下出現(xiàn)的幾個(gè)問題良风。
- 用vs的編譯工具編譯的時(shí)候提示error不支持預(yù)處理指令“warning”:
fatal error C1021: 無效的預(yù)處理器命令“warning”
根據(jù)pytorch論壇上的反饋,VS的版本太舊有可能是原因惩琉,但升級(jí)至最新版本VS后依然出現(xiàn)這個(gè)問題豆励。事實(shí)上,MSVC并不支持warning預(yù)編譯指令琳水,這一錯(cuò)誤的解決方法比較tricky肆糕,直接在pytorch源碼中注釋掉這一行預(yù)編譯指令即可般堆。
- 在后續(xù)編譯cu文件時(shí)出現(xiàn)不支持的編譯器版本問題在孝,查看相關(guān)cuda源碼即可發(fā)現(xiàn)cuda9.0源碼限制了MSVC編譯器的版本诚啃,根據(jù)pytorch論壇上的建議,VS一定要安裝15.3版本私沮。