본문 바로가기

딥러닝 프레임워크/PyTorch

PyTorch 설치 이후 Jupyter Notebook에서 import torch 안될때

* 문제: pytorch 설치 후 jupyter notebook에서 import torch가 실행이 안될 때

Anaconda를 설치할 때 Jupyter notebook이 같이 설치된다.

 

하지만 pytorch를 설치한 후, jupyter notebook을 실행해서

import torch

를 하면 torch가 import가 안되었다.

 

* 해결 방법

해결방법은 pytorch를 설치한 후, conda로 jupyter notebook을 다시 설치해주는 것이다.

conda install jupyter

기본적인 jupyter notebook 패키지는 anaconda에 들어가 있긴 하다.

하지만 pytorch 설치 후 jupyter notebook을 conda로 설치해주면 torch 환경을 jupyter notebook kernel에서 사용할 수 있게 된다.

 

여기서 순서가 pytorch 설치 후 jupyter notebook을 다시 설치해주는 것이다.

다시 설치해 주는 게, jupyter notebook을 pytorch를 포함한 이후의 anaconda env를 포함할 수 있도록 도와주는 것 같다. jupyter notebook을 install 하지만 update와 비슷한 의미인 것 같다.

 

다른 방법으로는 pytorch 설치 이후 jupyter notebook에 ipykernel을 추가하는 방법인 게 있는 것 같다.

 

즉 pytorch를 설치한 후의 anaconda env를 포함할 수 있도록 jupyter notebook을 업데이트 해주는 것이 중요한 것 같다.