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都能用,亲测有效。