딥러닝 프레임워크/PyTorch
PyTorch 설치 이후 Jupyter Notebook에서 import torch 안될때
keepee
2020. 3. 11. 23:08
* 문제: 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을 업데이트 해주는 것이 중요한 것 같다.