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

      在pycharm中import torch后,发现例如tensor等部分代码提示不再工作,网络上很多的解答都是基于低版本的torch。本人使用的pytorch版本是1.13.0,属于较新的版本,因此并不适用于这些方法。

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

         方法很简单,就是将上文提到的__init__.pyi文件复制到torch文件夹下 。对于我的项目具体的路径是D:\anaconda\envs\pyt\Lib\site-packages\torch,注意不要复制错位置。参考图如下 

关于pytorch在pycharm中部分代码提示失效的解决方法_第1张图片

关于pytorch在pycharm中部分代码提示失效的解决方法_第2张图片

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

__init__.pyi文件的分享见下面的百度网盘链接:

链接:https://pan.baidu.com/s/1c75e3xFhDi7VGh8Lsz_NsQ 
提取码:yhxq

你可能感兴趣的:(python,深度学习)