Mac에서 Pytorch 설정하기 (GPU 사용하기)

그저 간단한 Mac에서 Pytorch 설정

 

pip install torch
pip install torchvision

이 두개를 설치해주면 된다.

 

문제가 되었던 것은

 

Gpu를 사용할 때 였는데 Nvidia에서 GPU 사용에 필요한 MacOS 버전 CUDA를 지원을 안해주는 것이었다,,;

 

예제 실습하다가 왜 자꾸 "AssertionError: Torch not compiled with CUDA enabled "에러가 나서

 

CUDA 설치법 열심히 찾다가 Mac은 다르게 해야한다는 것을 알았다.

 

먼저 Mac에서 mps(Mac에서의 gpu? 인듯?) 사용할 수 있는지 보자

import torch

print (f"PyTorch version:{torch.__version__}")
print(f"MPS를 지원하도록 했는가?: {torch.backends.mps.is_built()}") 
print(f"MPS를 사용가능한가?: {torch.backends.mps.is_available()}")

둘 다 True로 나오면 된다고 한다.

 

보통 torch를 import 해주고 gpu를 어떤 것을 사용할지 device 변수에 넣어서 알려주는 것 같다.

import torch
import torchvision

device = torch.device('mps:0' if torch.backends.mps.is_available() else 'cpu')

device에 저렇게 넣어주자. "mps 디바이스 0번(첫번째) 거를 사용할 건데 사용 가능하면 mps를 쓰고 못쓰면 cpu를 사용해줘" 입니다.

 

뭐 간단하게 써도 사용은 되는 듯 하다.

device = torch.device("mps")