안녕하세요! 이번 포스팅에선 가상 환경을 이용한 pytorch 설치법에 대해 알아보고,
제가 겪은 import torch 에러(no module named torch)를 어떻게 해결했는지 알아보려 합니다.
먼저 가상 환경에 대해 알아보겠습니다.
가상 환경이란?
예를 들어 설명해보도록 하겠습니다.
만약 지금 파이썬 3.7 버전을 사용하고 있는데 3.6 버전을 이용할 일이 생겼습니다.
그렇다면 3.6버전을 이용하기 위해 기존의 3.7버전을 지우고 3.6버전을 다시 깔아야 할까요?
물론 그럴 수도 있겠지만 다양한 파이썬 프로젝트를 수행하다 보면 프로젝트마다 사용하는 버전이 다를 수 있고, 그때마다 지우고 다시 설치하는 것은 매우 번거로운 일일 것입니다.
이를 해결할 수 있는 방법이 가상 환경을 이용하는 것입니다.
가상의 환경을 2개 만들어 가상 환경 1에는 파이썬 3.6 버전을, 가상 환경 2에는 파이썬 3.7 버전을 설치하여 필요에 따라
환경만 바꿔가며 사용할 수 있을 것입니다.
이처럼 원하는 모듈이나 버전을 갖춘 가상의 환경들을 생성하여 독립적으로 관리하는 것입니다.
그럼 이제 pytorch를 설치할 가상의 환경을 만들고 설치를 진행해보도록 하겠습니다.
1. Anaconda Prompt를 우클릭하여 관리자 권한으로 실행시켜줍니다.
2. conda create -n 가상환경이름 을 이용하여 pytorch를 설치할 새로운 가상 환경을 만들었습니다.
(파이썬 버전 확인 결과 3.7.4이길래 python=3.7을 추가로 하여 가상 환경에서의 파이썬 버전을 3.7로 설정해주었습니다. 저는 이미 생성한 상태라 실행은 안 했지만 enter를 누르고 Proceed([y]/n)?이 나오면 y를 누르시면 됩니다!)
*참고 -> 가상환경 삭제 : conda remove -n 가상환경이름 --all
-conda env list / conda info --envs를 이용하면 가상 환경의 목록을 확인할 수 있습니다.
1)
2)
Pytorch라는 이름의 가상 환경이 잘 생성된 것을 볼 수 있습니다.
3. conda activate Pytorch -> 이제 가상 환경을 들어가 보겠습니다.
4. 이제 이 가상 환경에 pytorch를 설치해보도록 하겠습니다.
[pytorch 사이트] 에 들어가셔서 자신의 환경에 맞도록 설정해주시면 됩니다.
아래의 command를 cmd창에 입력하여 설치를 진행합니다.
- 체험판이 아닌 pytorch를 계속 쓸 거니까 preview가 아닌 stable로 깔아주고
- Anaconda를 이용하고 있으니 conda
- CUDA는 GPU 사용 여부와 관계가 있는데 NVIDIA의 그래픽 카드가 있어야 사용할 수 있습니다.
그래픽 카드를 확인하는 방법은 [장치 관리자] - [디스플레이 어댑터]
5. cmd 창에서 pytorch 잘 설치됐는지 확인
python
import torch
torch.rand(10)
코드를 실행하여 다음과 같이 잘 출력되면 설치가 잘 된 것입니다.
6. jupyter notebook import torch 에러
그럼 이제 pytorch를 사용해보기 위해 pytorch를 설치한 가상 환경에서 jupyter notebook을 실행시켜줍니다.
분명 설치를 제대로 해줬는데 jupyter notebook에서 실행이 되지 않습니다.
어떻게 된 일일까요?
conda list를 실행해주면 가상환경에 설치된 모듈을 볼 수 있습니다.
확인 결과 jupyter 모듈이 존재하지 않는 것을 확인했습니다.
새로 환경을 만들다 보니 기본이라 생각했던 것들이더라도 설치되어 있지 않은 것들이 많습니다.
따라서 가상 환경에서 conda install jupyter를 실행하여 설치해주었습니다.
제대로 설치됐는지 확인하기 위해 다시 conda list를 실행하여 모듈을 확인해봅니다.
잘 설치되었습니다.
이제 다시 가상 환경에서 jupyter notebook을 실행하여 torch가 잘 import 되는지 확인해보겠습니다.
import torch 에러가 해결되었습니다!
지금까지 가상 환경을 이용하여 pytorch를 설치해보았습니다.
이 때 pytorch나 jupyter는 conda activate 가상환경이름 을 실행하여 가상 환경에 들어간 후에 가상환경에 설치해야 된다는 점 유의하시기 바랍니다! 감사합니다 :D