完美(从根源上)解决Pytorch代码不提示问题

学习要深度

已于 2022-01-23 22:45:11 修改

阅读量2.6k

点赞数 1

文章标签： pytorch 深度学习人工智能

于 2022-01-23 22:34:53 首次发布

本文链接：https://blog.youkuaiyun.com/qq_41637232/article/details/122658030

版权

本文介绍了解决PyTorch 1.6.0版本中智能提示缺失的问题，通过对比1.8版本找到了关键差异，并提供了解决方案，包括修改源代码和升级版本。

摘要生成于 C知道，由 DeepSeek-R1 满血版支持，前往体验 >

作为面向IDE编程的程序员，没有代码提示能忍？

我的pytorch版本1.6.0，早在几个月前就发现了在敲torch.ones()、torch.zeros()这些常用的函数时没有智能提示，但一直没抽出时间来仔细研究这个问题，今天花了1个小时研究了一下，总算是解决了。
我在对比了1.8版本torch下的__init__().py文件时，终于发现了问题所在：
1.8.0

而我的1.6.0版本下的是：
1.6.0
从注释中可以大致看出，作者为了不让咱们在智能提示的时候显示"too many"的函数，直接把判断条件直接改成了False。而这个torch._C._VariableFunctions文件中全是abs、ones、zeros这样的常用函数，这就是无法智能提示的问题所在！

而1.6.0与1.8.0版本的不同之处，就是当中的那个变量TYPE_CHECKING。ctrl点进去后，我就发现了问题：

这个TYPE_CHECKING是一个动态的值，当你敲代码时，它为True，从而能够导入torch._C._VariableFunctions，进而智能提示。

经过这么一通分析，方法就很简单了：
找到torch下__init__().py中的这句代码

from typing import Set, Type

在其后面添加一个变量：TYPE_CHECKING
然后把下面这行代码中的的False改为TYPE_CHECKING就OK啦

if False:
    # Some type signatures pulled in from _VariableFunctions here clash with
    # signatures already imported. For now these clashes are ignored; see
    # PR #43339 for details.
    from torch._C._VariableFunctions import *  # type: ignore