PyTorch在Pycharm等IDE中提示不完全等問題,想必搞深度學習/AI的人甚爲之頭疼。而且你查資料,各種風騷操作,但最後,基本上,一個個都是:
不能用!!!
連谷歌也搜不到管用的東西。
樓主也搜索了很多資料,然後丟掉。自己想了一想,方法其實很簡單:
按理說PyTorch 在1.0以後修復了此Bug,我覺得Facebook的那幫人測試做得賊不好。可能測試都是外包出去的。畢竟老美資本主義爲了節省成本啥都幹得出來。不然波音會把重要的系統外包出去,出現兩次空難死幾百人麼? 而且測試都是大佬們不願意乾的事情,錢少,而且不高大上。所以常常出問題。
問題主要在於__init__.pyi文件,打開這個文件,發現各類的函數都導進來了吶?有兩種:
下載見https://download.csdn.net/download/dongfangxiaozi_/11460709
1.本地的話,直接把最新的__init__.pyi覆蓋掉。記得提前備份。
2.遠程服務器的話,發現只有__init__.py,沒有__init__.pyi,這時候,把__init__.pyi放在__init__.py旁邊,立馬好了。
如何找__init__文件在哪裏,在IDE裏,import torch, 然後點進去torch,再右鍵在文件夾打開,無論是Windows/MacOS/linux都能用,親測有效。