tensorflow 설치 _ GPU 사용

Command

# 20221219 - MacOS
% conda install -c apple -y tensorflow-deps==2.9.0
% pip install tensorflow-macos==2.9.0
% pip install tensorflow-metal==0.5.0

Step 1: Xcode command line tools 설치하기

% xcode-select --install

Step 2: Miniforge 설치하기

% brew install --cask miniforge

Step 3: Conda를 이용해 가상환경 만들기

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

Step 4: Tensorflow-MacOS 설치하기

# tensorflow의 의존성을 설치한다.
% conda install -c apple -y tensorflow-deps
 
 
# tensorflow의 본 패키지를 설치한다. 
% pip install tensorflow-macos
 
 
# metal 플러그인(GPS 사용)을  설치한다
% pip install tensorflow-metal

Step 5: GPU의 감지 여부와 사용가능한 GPU의 개수 확인

# 다음으로 GPU의 감지 여부와 사용가능한 GPU의 개수를 확인해 본다. 
>>> import tensorflow as tf
>>> print("Num GPUs Available: ", len(tf.config.experimental.list_physical_devices('GPU')))

Reference

출처