Apple M1 GPU 사용하는 방법

PyTorch로 GPU 설정

반응형

## PyTorch 설정
import torch

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

print (f"PyTorch version:{torch.__version__}") # 1.12.1 이상
print(f"MPS 장치를 지원하도록 build 되었는지 확인: {torch.backends.mps.is_built()}") # True
print(f"MPS 장치가 사용 가능한지 확인: {torch.backends.mps.is_available()}") # True
!python -c 'import platform;print(platform.platform())'

 

결과

 

Tensorflow로 GPU 설정

## tensorflow GPU 설정
import tensorflow as tf
print("Num GPUs Available: ", len(tf.config.experimental.list_physical_devices('GPU')))