[환경설정] 맥에서 Anaconda 환경으로 pytorch 사용하기

졸업 작품으로 pytorch를 이용하여 image classification을 할 일이 생겨

Anaconda를 사용하여 맥에서 pytorch를 사용해보려 합니다.

1. Anaconda 설치

운영체제에 맞는 Anaconda installer를 다운 받아 설치해줍니다.

Anaconda | Individual Edition

Anaconda's open-source Individual Edition is the easiest way to perform Python/R data science and machine learning on a single machine.

www.anaconda.com

이때 터미널에서 conda라는 명령어를 쳐보면 해당 명령어가 없다는 에러가 뜨게 됩니다.

2. 환경변수 설정

~/.zshrc(또는 ~/.bashrc) 파일을 다음과 같이 작성해 줍니다.

(터미널에서 vi ~/.zshrc 명령으로 들어가 수정할 수 있습니다.)

# ~/.zshrc
export PATH="/opt/anaconda3/bin:$PATH"

터미널을 재실행하거나 source ~/.zshrc 명령을 통해 스크립트를 실행합니다.

3. 가상환경 생성 및 활성화

# 가상환경 생성 (--name 원하는 가상환경 이름)
conda create --name pytorch_env python=3

# 가상환경 활성화
conda activate pytorch_env
# source activate pytorch_env

# 비활성화
# conda deactivate

가상환경을 생성하고 활성화 해줍니다.

4. 패키지 설치

conda install -y pytorch torchvision -c pytorch
conda install -y numpy

필요한 패키지들(pytorch, torchvision)을 활성화한 가상환경에 설치합니다.

만약 패키지가 없다고 나오면 update 명령어로 conda를 업데이트 시켜줍니다.

conda update -n base -c defaults conda

5. pytorch import

파이썬 파일을 생성하고 import를 했을 때 정상적으로 동작하는지 확인합니다.

import torch
import torchvision
import numpy as np

leettle log