read

PyTorch 설치가 어려울 때, conda 가상 환경 안에 PyTorch를 설치하면 깔끔하게 설치될 때가 많습니다. 이 글은 conda 가상 환경으로 PyTorch를 설치하고 Jupyter의 kernel로 등록하는 방법을 소개합니다. TensorFlow도 같은 방법으로 설치할 수 있습니다.

Windows

새 가상 환경 만들기

$ conda create -y -n pytorch ipykernel

pytorch 대신 자신이 원하는 이름을 쓸 수 있습니다.

가상 환경 안으로 들어가기

$ activate pytorch

PyTorch 설치하기

(pytorch)$ conda install -y -c peterjc123 pytorch

Jupyter에 새 kernel 등록하기

(pytorch)$ python -m ipykernel install --user --name pytorch --display-name "PyTorch"

--display-name은 Jupyter Notebook 위에서 표시될 kernel의 이름으로 "PyTorch" 대신 자신이 원하는 이름을 쓸 수 있습니다.

가상 환경 빠져나오기

(pytorch)$ deactivate

MacOS / Linux

새 가상 환경 만들기

$ conda create -y -n pytorch ipykernel

pytorch 대신 자신이 원하는 이름을 쓸 수 있습니다.

가상 환경 안으로 들어가기

$ source activate pytorch

PyTorch 설치하기

(pytorch)$ conda install -y pytorch torchvision -c pytorch

Jupyter에 새 kernel 등록하기

(pytorch)$ python -m ipykernel install --user --name pytorch --display-name "PyTorch"

--display-name은 Jupyter Notebook 위에서 표시될 kernel의 이름으로 "PyTorch" 대신 자신이 원하는 이름을 쓸 수 있습니다.

가상 환경 빠져나오기

(pytorch)$ source deactivate

Jupyter에 등록된 kernel 확인하기

이제 Jupyter Notebook에서 새 노트북을 만들 때 PyTorch kernel을 선택할 수 있습니다.

Jupyter Notebook에 등록된 PyTorch kernel

해설

새 가상 환경 만들기에 대해서

Conda는 패키지 관리 프로그램인 동시에 가상 환경 관리 프로그램입니다. Conda로 기존 환경과 충돌이 없는 가상 환경을 만들고 관리할 수 있습니다.

$ conda create -y -n pytorch ipykernel
  • conda create : 새 conda 환경을 만듭니다.
  • -y : --yes의 줄임말입니다. 설치 승인을 생략하고 바로 설치합니다.
  • -n pytorch : --name pytorch의 줄임말입니다. 환경 이름을 pytorch로 짓습니다. 환경 이름은 필수입니다. 이름은 원하는 대로 지을 수 있습니다.
  • ipykernel : ipykernel이 설치되어 있는 가상 환경을 만듭니다. ipykernel 외에 다른 패키지 이름을 쓸 수 있습니다.

Jupyter에 새 kernel 등록하기에 대해서

(pytorch)$ python -m ipykernel install --user --name pytorch --display-name "PyTorch"
  • python -m ipykernel : ipykernel 모듈을 파이썬 스크립트로 실행합니다.
  • —name pytorch : Jupyter 내부적으로 쓰이는 kernel의 이름을 지정합니다. 같은 이름을 쓰면 덮어쓰기가 됩니다.
  • --display-name "PyTorch" : Jupyter Notebook 위에서 사용자에게 보이는 kernel의 이름을 정합니다. 내부적으로 쓰이는 이름과 상관없이 띄어쓰기 등 특수문자도 포함하여 자유롭게 지을 수 있습니다.

유용한 Conda 명령어

conda를 사용할 때 자주 사용하는 명령어들을 모아보았습니다.

가상 환경 만들기 옵션

파이썬만 있는 최소한의 환경을 원할 때

$ conda create --name myenv python

파이썬과 pip 등의 최소한의 패키지만 설치되어 있는 myenv라는 이름의 새 가상 환경을 만듭니다. ipykernel, numpy 등이 모두 없는 환경입니다. myenv 대신 원하는 이름을 쓸 수 있습니다.

파이썬 버전을 지정하고 싶을 때

$ conda create --name myenv python=2.7

Python2.7의 가상 환경을 만듭니다. 이처럼 원하는 파이썬 버전을 지정하여 가상 환경을 만들 수 있습니다. 위와 마찬가지로 최소한의 환경만 설치합니다.

아나콘다 환경을 만들고 싶을 때

$ conda create --name myenv anaconda

아나콘다 환경에는 numpy, pandas 등 수학/과학 관련 패키지들이 포함됩니다. ipykernel 역시 설치되어 있습니다.

파이썬 버전을 지정하고 아나콘다 환경을 만들고 싶을 때

$ conda create --name myenv python=2.7 anaconda

가상 환경 목록 보기

$ conda env list

pytorch
myenv
myenv2
...

가상 환경 삭제하기

$ conda env remove --name myenv

참고 자료

Blog Logo

Dreamgonfly


Published

Image

Data Storyteller

Dreamgonfly's blog

Back to Overview