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

polarcompass 2021. 10. 22. 14:10

728x90

1. Conda env 설치

$ brew install miniforge

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

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

출처 동영상 ※ 기본적으로 위 동영상을 시청하면서 따라하시는것을 추천드립니다. 아래 코드블럭으로 표현 못하는 경로 찾는 방법들이 후반에 많이 나오기에 영상을 보면서 따라해주세요. 1. X

polarcompass.tistory.com

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

Metal - Apple Developer

Find presentations, documentation, sample code, and resources for building macOS, iOS, and tvOS apps with the Metal framework.

developer.apple.com

728x90