解决pytorch在pycharm中某些函数不能自动提示

好久没有写博客了，虽然不是大佬，但是还是可以一点一点记录一下的，毕竟很多都是自己折腾自己踩得坑，值得记录一下。争取硕士毕业进入博客前10000名哈哈，还有三个多月。

---------------------------------------------------------------------

今天主要记录下Torch在pycharm中无法自动提示。

网上的答案比较少，或者许多人没有遇到这个问题？

之前Torch的版本在一些IDE当中确实无法自动提示的，解决方法是GitHub上的一个__init__.pyi文件，可以解决。

 

不过现在比较新的Torch版本是支持自动提示的，比如1.0以上。

而我在安装Torch 1.0.1 后，在pycharm中打代码，某些函数还是不能提示，情况是这样的。

如上图，我在巧torch.nn或者torch.optim时候，不能自动提示。

一直把问题想复杂了，以为不能自动提示，一直谷歌找答案。直到某次敲代码的时候，突然发现可以自动提示了，然后阅读了一下代码，发现只要在之前导入就行了。

OK了，解决了疑惑，可以舒服地打代码了，不然心里有个结。