TensorFlow tutorial

python 프로그래밍 - TensorFlow

Introduction

Google 에서 2015년에 공개한 오픈소스 라이브러리이다. 대량의 수치 계산을 효과적으로 지원하기 위해 개발되었지만 지금은 기계학습과 딥러닝 모델을 구현하고 실행하는데 널리 사용된다.

m1 mac에 tensorflow 설치

1. 파이썬 버전이 3.9보다 낮거나 같아야하므로 가상환경을 만들었다.

   $ pip install virtualenv
   $ virtualenv tf-tutorial -p python3.9
   $ source path/to/venv/bin/activate
   

2. base TensorFlow 설치

   $ python -m pip install tensorflow-macos
   

3. tensorflow-metal 플러그인 설치

   $ python -m pip install tensorflow-metal
   

   Metal

   Metal이란 애플에서 만든 그래픽 API이다. mac에서 동작하는 앱들을 위해 그래픽 하드웨어에 접근할 수 있는 API를 제공한다. 쉽게 말하면 macOS에 최적화된 OpenGL이다.

4. 잘 설치되었는지 확인

   아래의 스크립트 실행하면 된다.

    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=True)
    model.compile(optimizer="adam", loss=loss_fn, metrics=["accuracy"])
    model.fit(x_train, y_train, epochs=5, batch_size=64)
   

   결과

reference

• Get started with tensorflow-metal