맥OS Apple Silicon 텐서플로우 메탈 PluggableDevice 개발환경 설정 tensorflow-metal PluggableDevice

1. Conda env 설치

※ miniforge 환경에서 설치한다!!!

$ brew install miniforge

아래 링크로 들어가서 conda env 설치까지 해주고 다시 이 페이지로 돌아온다.

https://polarcompass.tistory.com/24?category=512101

맥북 M1 OpenCV 설치 환경 설정 개발 환경 세팅 Big Sur AppleSilicon chipset 애플 실리콘 칩셋

 

2. TensorFlow dependencies 설치

2-1. 먼저 설치된 텐서플로우를 지워준다.

# uninstall existing tensorflow-macos and tensorflow-metal
python -m pip uninstall tensorflow-macos

python -m pip uninstall tensorflow-metal

2-2. tensorflow-deps로 업그레이드 해준다.

conda install -c apple tensorflow-deps --force-reinstall

※가상환경 내에서 설치가 안될시 가상환경을 지명해서 설치해준다.

conda install -c apple tensorflow-deps --force-reinstall -n <my_env>

맨끝의 <my_env>를 내가 설정한 가상환경 폴더명을 지명해서 설치해준다.

2-3. tensorflow-deps 원하는 version을 선택해서 설치해준다.

● v2.5

conda install -c apple tensorflow-deps==2.5.0

● v2.6

conda install -c apple tensorflow-deps==2.6.0

3. base TensorFlow 설치해준다.

python -m pip install tensorflow-macos

4. tensorflow-metal plugin 설치

python -m pip install tensorflow-metal

5. keras  사용시 자동완성 적용 방법

import tensorflow.keras as keras

위와 같이 적용시 vscode에서 자동완성 적용 가능

Reference

https://developer.apple.com/metal/tensorflow-plugin/?fbclid=IwAR20gvrmhi5tBdPr2YX4iVtE_G6Bnuv1Zh1Rush5aFgfrJjjNGVYw1JIr8c

Metal - Apple Developer