맥북 m1 Conda 환경에 Tensorflow 설치하기

박태윤·2022년 3월 6일
1
post-thumbnail

M1 맥에 TensorFlow 설치하기 (macOS 12.0 이상)

Xcode Command Line Tools 설치

% xcode-select --install

Miniforge

% chmod +x ~/Downloads/Miniforge3-MacOSX-arm64.sh
% sh ~/Downloads/Miniforge3-MacOSX-arm64.sh
% source ~/miniforge3/bin/activate

애플 사이트에서는 위와 같은 방법으로 Miniforge를 설치하도록 하고 있는데, 아래 11.0 이하처럼 brew를 사용해도 무방하다.

가상환경 만들기

Conda를 이용한다는 것은 가상환경을 이용한다는 것이니, Tensorflow를 설치할 가상환경을 만들어주자.

% conda create --name DL-tf python=3.8
% conda activate DL-tf

TensorFlow Dependencies 설치

# For TensorFlow v2.5
% conda install -c apple tensorflow-deps==2.5.0

# For TensorFlow v2.6 
% conda install -c apple tensorflow-deps==2.6.0

TensorFlow & Plugin 설치

% python -m pip install tensorflow-macos
% python -m pip install tensorflow-metal

M1 맥에 TensorFlow 설치하기 (macOS 11.0 이하)

Xcode Command Line Tools 설치

% xcode-select --install

Brew 설치

패키지 관리자 Brew를 설치한다!

% /bin/bash -c "$(curl -fsSL https://raw.githubusercontent.com/Homebrew/install/HEAD/install.sh)"

Miniforge 설치

방금 설치한 Brew를 이용해 Miniforge를 설치한다.

Miniforge는 맥용 arm64, 즉 M1 맥을 지원하고 있는 conda라고 보면 된다. Anaconda 등은 M1 맥을 지원하고 있지 않다. 사용 자체는 되는데, Rosetta 2를 통한 x64 패키지들을 설치한다.

% brew install cask
% brew install --cask miniforge
% conda init zsh

이렇게 설치한 miniforge의 설치 경로는 다음과 같다.

/opt/homebrew/Caskroom/miniforge

가상환경 만들기

Conda를 이용한다는 것은 가상환경을 이용한다는 것이니, Tensorflow를 설치할 가상환경을 만들어주자.

% conda create --name DL-tf python=3.8
% conda activate DL-tf

–name 뒤의 DL-tf는 원하는 대로 정해도 좋다.

패키지 설치

이후 tensorflow 설치에 필요한 패키지를 설치한다.

 % conda install absl-py astunparse gast google-pasta grpcio h5py=2.10.0 ipython keras-preprocessing numpy=1.19.5 opt_einsum pip=20.2.4 protobuf python-flatbuffers scipy tensorboard tensorflow-estimator termcolor typeguard typing_extensions wheel wrapt

Tensorflow 설치

pip를 이용해 tensorflow를 설치한다.

% pip install --upgrade --force --no-dependencies https://github.com/apple/tensorflow_macos/releases/download/v0.1alpha3/tensorflow_macos-0.1a3-cp38-cp38-macosx_11_0_arm64.whl https://github.com/apple/tensorflow_macos/releases/download/v0.1alpha3/tensorflow_addons_macos-0.1a3-cp38-cp38-macosx_11_0_arm64.whl

설치 확인

% conda activate DL-tf
% python3 -c "import tensorflow as tf; print(tf.reduce_sum(tf.random.normal([1000, 1000])))"

결과가 tf.Tensor(-771.1804, shape=(), dtype=float32) 형태로 나오면 import 하고 실행하는데 성공한 것이다.

Reference

profile
이제 시작해 봅시다.

0개의 댓글