[Mac] Tensorflow 설치

▷M2 Macbook Air

이제 Macbook산지 1달이 넘어가는... 드디어 Tensorflow랑 Pytorch를 설치해보고자한다. 에어가 견뎌주길...핳

오늘은 Tensorflow설치하는 과정을 정리했다.

이전에 Anaconda를 다 설치해놓아서 파이썬 버전만 맞추니 금방 깔렸다!

 

1. Anaconda 설치

아직 설치 전이라면 전에 작성한 Anaconda설치 바로가기 보고 참고하면 된다!

 

2. 가상환경 생성(파이썬 버전은 3.8이나 3.9로 설치!)

https://www.tensorflow.org/install?hl=ko - Python 3.6~3.9로 나와있기 때문에.. 근데 또 아래 metal에서는 3.8이나 이후버전으로 설치하라고한다. 그래서 파이썬 버전은 3.8이나 3.9로 설치하는 걸루..3.12 버전했다가 에러가 났다ㅠ

conda create -n 가상환경이름 python=버전

      저는 3.8/3.9버전 다 했는데 잘됐습니다ㅎ

 

3. 가상환경 활성화

conda activate 가상환경이름

 

4. Tensorflow 설치

https://developer.apple.com/metal/tensorflow-plugin/ - 참고

Tensorflow Plugin - Metal - Apple Developer

> 가상환경 활성화된 상태로 입력한다!! (3번과정 꼭!!)

python -m pip install tensorflow          #TensorFlow 버전 2.13 이상의 경우

python -m pip install tensorflow-metal    #Tensorflow-metal 플러그인 설치

 

5. 실행

> vscode에서 아래코드 복붙한 py파일 만들고 fn+f1키 눌러서 Python:select interpreter를 위에서 생성한 가상환경으로 선택해 실행!

import tensorflow as tf

cifar = tf.keras.datasets.cifar100
(x_train, y_train), (x_test, y_test) = cifar.load_data()
model = tf.keras.applications.ResNet50(
    include_top=True,
    weights=None,
    input_shape=(32, 32, 3),
    classes=100,)

loss_fn = tf.keras.losses.SparseCategoricalCrossentropy(from_logits=False)
model.compile(optimizer="adam", loss=loss_fn, metrics=["accuracy"])
model.fit(x_train, y_train, epochs=5, batch_size=64)

코드 돌아가는 동안 활성상태보기 누르면 GPU값이 확 올라가는 것을 볼 수 있다.

에러안나고 실행완료~~~~~~

 

> 버전은 터미널에 conda list쳐서 확인!

  

 

에어로 돌려서 발열이 되긴하지만(계속 불안해하면서 epoch2번할때 끊을까 생각했돈ㅋㅋ 발열은 심해도 잘 됐댜 휴..)

속도도 꽤 빠르고 Gpu를 쓴다는점에 ㅎㅎ

이제 Pytorch도 설치하구 간단한^^ 딥러닝코드돌려보자명!!!!(맥북 에어는...)