[pycharm] pytorch intellisense

pytorch를 pycharm에서 사용할 때 특정 모듈은 인텔리센스에서 뜨질 않는다. Python Console에서는 잘 뜨는데 그 이유는 Python Console은 실제로 import를 했기 때문에 구문 분석이 가능하지만 Editor에서는 실제로 import를 하지 않았기 때문에 특정 모듈이 인텔리센스에 잡히지 않는 것이다. 예를 들면 torch.utils, torch.nn.functional 등은 인텔리센스에 잡히지 않는다.

간단하게 인텔리센스에 넣는 방법은 아래처럼 그냥 해당 모듈을 찾아 들어가 임포트해주면 된다.

import torch.utils.data
import torch.nn.functional

References

• https://intellij-support.jetbrains.com/hc/en-us/community/posts/360000142284-PyChram-autocomplete-for-import-library
• https://discuss.pytorch.org/t/torch-utils-has-no-module-data/1712