AI·머신러닝·딥러닝

macOS Apple Silicon m1/m2/m3에서 Tensorflow GPU 환경 구성하기

무한열정 2024. 3. 5. 10:12

■ 기본 설치환경

mac mini m2 깡통(8Core CPU + 10Core GPU) + 16G RAM 

macOS Sonoma 14.3.1

Anaconda v2023.09-0 (Python 패키지 관리)
Python v3.11
Tensorflow-macos v2.15.0
Tensorflow-metal v1.1.0

 

 설치후 결론

- Linux 환경에서 CUDA 설치가 까다로운데 비해 macOS Metal 기반 Tensorflow는 설치가 아주 쉬웠다. ( pip로 간단하게 설치 완료)

 

 설치 방법

 

Anaconda VM 환경 추가 ( Python v3.11 기반 )

$ conda create -n tf-mac-metal python=3.11

 

Anaconda 생성한 환경으로 전환 (24-03-05 현재 v1.1.0이 설치됨)

conda activate tf-mac-metal

 

Tensorflow-metal 설치 (24-03-05 현재 v2.15.0이 설치됨)

$ pip install tensorflow-metal

 

Tensorflow-macos 설치

$ pip install tensorflow-macos

 

Tensorflow GPU 사용 확인하기

$ python -c "import tensorflow as tf; print(tf.config.list_physical_devices('GPU'))"

[PhysicalDevice(name='/physical_device:GPU:0', device_type='GPU')]

 

Nvidia CUDA 및 cuDNN 환경에 비하면 비교도 안 되게 아주 쉽게 설치가 완료된다. ^^

 

 

 Tensorflow-macos 및 Tensorflow-metal 호환성

https://pypi.org/project/tensorflow-metal/