关于pytorch在pycharm中部分代码提示失效的解决方法

本文转载自：https://blog.csdn.net/snowbast/article/details/127624339

问题

在pycharm中import torch后，发现例如tensor等部分代码提示不再工作，比如以下代码基本都是没有代码提示:

a = torch.rand(2,2,3)
print(a.shape)
print(a.size())
print(a.numel())
print(a.dim())

网络上很多的解答都是基于低版本的torch。本人使用的pytorch版本是1.13.0，属于较新的版本，因此并不适用于这些方法。

因为之前也有遇到过opencv失去智能提示的问题再结合上在github上看到过某大佬发布的__init__.pyi文件。因此便萌生了以下想法，当然也顺利解决了这个问题。

解决方法

方法很简单，就是将上文提到的__init__.pyi文件复制到torch文件夹下。

__init__.pyi 文件下载链接：https://files.cnblogs.com/files/FengZeng666/pyi.zip

对于我的项目具体的路径是E:\ProgramSoftware\Python\anaconda3\Lib\site-packages\torch，注意不要复制错位置。参考图如下：

复制结束后关闭pycharm，重新打开后就会发现代码提示恢复正常了。