M1 맥에서 텐서플로(Tensorflow) GPU를 쓰고 싶다.

텐서플로 역시 파이토치(Pytorch)와 같이 설치가 간단하다.

여기서 텐서플로는 2.13 버전 이상이다.

# conda create -n tf_py39 python=3.9 -y
# conda activate tf_py39

$ pip install tensorflow

 

텐서플로에서 활성화할 수 있는 물리적 장치 목록을 보는 방법은 간단하다.

import tensorflow as tf

# List available devices and print them
devices = tf.config.list_physical_devices()
for device in devices:
    print(device)

 

비교적 간단하게 설치했지만서도 장치 목록에 CPU만 보이는 상황,

어쩐지 쉽더라니..

 

텐서플로를 설치한 후 tensorflow-metal 라이브러리를 추가로 설치해줘야한다.

$ python -m pip install tensorflow-metal

 

설치가 잘 되었다면, 장치 목록에서 GPU가 보일 것이다.

(주피터 노트북 환경을 쓰고 있다면 커널을 재시작 해줘야한다.)

일반적으로 GPU 라 함은 NVIDIA의 그래픽카드를 이야기하고, 곧 CUDA 사용을 의미한다.

M1 맥북에는 MPS라고 하는 GPU 가속화 기능이 있어서 이를 사용할 수 있다.