[Arm64] M1 맥북에 tensorflow 2.4 설치하기 (conda 환경)

[Arm64] M1 맥북에 tensorflow 2.4 설치하기 (conda 환경)

출처: 참고한 글 

 

 

문제 인식 

 

몇달전 M1 맥북을 사고 머신러닝 작업을 위해 tensorflow를 사용해 보았다. 쥬피터 노트북 (jupyter notebook) 으로 작업하고 있는데, 이상하게 tensorflow를 import만 하면 "The kernel appears to have died. It will restart automatically" 라는 오류를 내며 JN이 죽어버리는 현상에 직면하게 되었다. 

 

이것을 해결하려고 한참 동안 검색을 하다 알게 된 것이 바로 python과 tensorflow의 버전이 맞아야 한다는 것이다. "tensorflow python version compatibility"로 검색하면 그 정보를 찾을 수 있다. 

 

https://www.tensorflow.org/install/pip

나의 python 버전은 3.8이었고, tensorflow는 2.0을 사용하고 있었기에 버전 호환이 되지 않아 import tensorflow에서 커널이 죽는 것이었다. 

 

아나콘다에서 지원하는 텐서 플로우의 최신 버전은 2.0까지여서, pip으로 2.4로 업그레이를 해주어 봤지만 문제가 해결되지 않았다. 그 이유는 pip을 통해 내려받을 수 있는 tensorflow 2.4는 arm64를 지원하지 않기 때문이었다. 

 

conda search tensorflow 

 

따라서, m1 맥용 tensorflow 2.4를 수동으로 설치해 주어야 한다. 다음의 링크에서 그 방법을 찾을 수 있었고 문제를 해결했다. 

(참고한 글) TensorFlow 2.4 on Apple Silicon M1: installation under Conda environment

 

이 경험을 공유하고자 포스팅을 해본다. 

 

 

문제 해결: Arm64 TensorFlow(ATF) 2.4 설치 (alpha 3.0)

 

1. Xcode command line tools 설치 

xcode-select --install

 

2. miniforge 설치 

Install miniforge for arm64 (Apple Silicon) from miniforge github.

 

bash Miniforge3-Linux-x86_64.sh

 

3. ATF2.4 내려받기 

github.com/apple/tensorflow_macos/releases 이 링크에서 Assets 메뉴를 펼치면 내려받을 수 있는 파일들이 있다. 

'tensorflow_macos-0.1alpha3.tar.gz' 을 받아 압축을 풀고 아래와 같이 하위 폴더에서 파일을 확인한다. 

cd tensorflow_macos/arm64

arm64 디렉토리 내용

 

4. Conda 환경 설정 

conda create --name tf24
conda activate tf24
conda install -y python==3.8.6
conda install -y pandas matplotlib scikit-learn jupyterlab

 

5. ATF 2.4 라이브러리 설치 

# Install specific pip version and some other base packages
pip install --force pip==20.2.4 wheel setuptools cached-property six packaging

# Install all the packages provided by Apple but TensorFlow
pip install --upgrade --no-dependencies --force numpy-1.18.5-cp38-cp38-macosx_11_0_arm64.whl grpcio-1.33.2-cp38-cp38-macosx_11_0_arm64.whl h5py-2.10.0-cp38-cp38-macosx_11_0_arm64.whl

# Install additional packages
pip install absl-py astunparse flatbuffers gast google_pasta keras_preprocessing opt_einsum protobuf tensorflow_estimator termcolor typing_extensions wrapt wheel tensorboard typeguard

# Install TensorFlow
pip install --upgrade --force --no-dependencies tensorflow_macos-0.1a3-cp38-cp38-macosx_11_0_arm64.whl
pip install --upgrade --force --no-dependencies tensorflow_addons_macos-0.1a3-cp38-cp38-macosx_11_0_arm64.whl

 

6. ATF 2.4 설치 확인 

 

 

happy native machine learning on Apple silicon!!! 

 

 

2021.08.15 - [Programming/Tips] - [Arm64] M1 맥북에 tensorflow 2.5 설치하기 (M1 GPU 사용가능)

[Arm64] M1 맥북에 tensorflow 2.5 설치하기 (M1 GPU 사용가능)