아나콘다에서 PyTorch 설치, PyCharm에서 돌려보기

솔직히 PyTorch를 쓰려면 GPU를 통해 연산을 처리해야 하는데, 지원하는 API가 CUDA이다.

하지만 NVIDIA GPU가 있어야 CUDA를 설치할 수 있고 내 컴퓨터는 인텔 그래픽카드가 내장되어 있어서 아쉽지만 OpenCL을 지원하는 다른 딥러닝 프레임워크를 사용할까 생각 중이다.

그래도 일단 한번 설치해보았다.

 

설치하기 이전에, 파이썬과 아나콘다를 이미 설치해두었다.

당연히 파이썬은 미리 설치를 해야 하고 (또는 C++/Java가 지원되거나), 아나콘다는 굳이 사용하지 않아도 cmd에서 pip명령어를 사용해서 다운로드할 수도 있다.

 

1. 아나콘다에서 가상 환경을 만들어준다.

이거 안 만들었다가 다시 삭제하고 실행했으니 꼭 환경 먼저 만들어주도록 하자.

conda create -n pytorch python=3.6
activate pytorch

2. 그리고 파이 토치 사이트에 들어간다.

3. Install PyTorch부분을 보면 선택하는 부분이 있는데

Stable(1.7.0) - Windows(운영체제) - Conda(패키지) - Python(언어) - None(CUDA)를 선택했다.

아래에 커멘드가 뜨면 복사해서 Anaconda에서 돌리면 된다.

만약 같은 조건이라면 아래의 코드를 그대로 돌리면 된다.

conda install pytorch torchvision torchaudio cpuonly -c pytorch

4. 아나콘다 안에서 파이썬을 실행하고 바로 돌려보자.

python
>>> import torch
>>> cpu_tensor = torch.zeros(2, 3)
>>> device = torch.device("cpu")
>>> tensor = cpu_tensor.to(device)
>>> print(tensor)

결과

tensor([[0., 0., 0.],
        [0., 0., 0.]])

5. PyCharm에서도 돌리고 싶다면 File-NewProject 클릭

설치는 여기에서 community 버전을 다운받으면 된다.

동영상 가이드는 여기에...

파일 위치는 Users\...\anaconda3\envs\pytorch

그리고 파일 이름 적어주기 Users\...\anaconda3\envs\pytorch\NewProjectName

또 Existing Interpreter을 쓴다고 선택하고 우측 점 3개 클릭

6. Conda Environment 선택 후 우측 파일 아이콘 선택

그리고 아까와 같은 경로로 간 후 pytorch폴더의 python.exe선택

7. 새로 생긴 프로젝트에서 터미널을 열어서 (프로젝트 폴더 우클릭 -> Open in Terminal)

아까처럼 python실행 후 import torch 실행해서 문제없으면 성공이다 :)