本文轉載自：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，重新開啟後就會發現程式碼提示恢復正常了。