關於pytorch在pycharm中部分代碼提示失效的解決方法 關於pytorch在pycharm中部分代碼提示失效的解決方法

關於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，重新打開後就會發現代碼提示恢復正常了。

---