아나콘다 GPU 환경 설정(python3.9, torch1.11.0)

 

아나콘다 설치 방법은 여러 곳에서 찾아볼 수 있으니 생략하겠다. GPU를 사용하는 머신러닝을 위해서는 환경설정이 중요하다. 아나콘다를 이용해 환경 세팅을 해주어야 하는데 파이썬, 파이토치, 쿠다 버전을 맞추기가 까다롭다. 각 버전별로 호환이 되는 것들이 정해져있어서 다른 버전으로 다운받았다가는 싹 다 엎고 재설치를 해야 한다.

여기서는 python3.9에 pytorch를 설치하여 GPU 환경을 구축하는 방법을 소개한다. 먼저 아래와 같은 커맨드로 python3.9를 사용하는 콘다 가상환경을 생성한다.

conda create -n [환경이름] python=3.9

 

가상환경이 만들어졌으면 아래 커맨드로 해당 가상환경을 활성화시킨다.

conda activate [환경이름]

 

그 후 아래 커맨드로 python3.9에 호환이 되는 torch, cuda를 비롯한 패키지들을 설치한다.

conda install pytorch==1.11.0 torchvision==0.12.0 torchaudio==0.11.0 cudatoolkit=11.3 -c pytorch