M1 칩 Mac에서 tensorflow 사용하기

나는 pytorch만 쓰는데 주변에서 mac에서 tensorflow가 안된다고 해서 세팅해줬다. 

 

1. Anaconda 환경 설정

conda create --name tf_env python=3.9
conda activate tf_env

먼저 conda 가상환경을 세팅하자

 

2. Tensorflow 및 패키치 설치

conda install -c apple tensorflow-deps

pip install tensorflow-macos tensorflow-metal

tensorflow-metal은 apple의 metal API를 통해 GPU 가속을 가능하게 해주는 플러그인이다.

 

3. Jupyter lab에서 환경 설정 (optional)

conda install -c conda-forge jupyterlab

pip install ipykernel
python -m ipykernel install --user --name=tf_env --display-name "Python (tf_env)"

jupyter lab 사용하면 새 환경을 jupyter lab의 커널로 추가한다.

 

위와 같이 몇 가지 세팅을 하면 tensorflow를 mac에서도 사용할 수 있다. 

 

 

근데 웬만하면 torch 쓰길 ㅎㅎ