Swift for TensorFlow で MNIST を実行する

Swift for TensorFlow

少し前に TensorFlow Dev Summit 2018 でアナウンスされた Swift for TensorFlow の動画は以下

Swift for TensorFlow - TFiwS (TensorFlow Dev Summit 2018)

ついにそれが先週が OSS になった

https://github.com/tensorflow/swift https://www.tensorflow.org/community/swift

色々とすごいところが多いのだけれど、このブログでは Swift for TensorFlow を動かしてみるところに注力してみる

Swift for TensorFlow のインストール

ビルドされた TensorFlow が入ってる Swift を以下のリンク先の .dmg からインストールする

https://github.com/tensorflow/swift/blob/master/Installation.md#pre-built-packages

PATH を通す

コマンドラインからビルドするだけなのでパスを通すだけで OK

export PATH=/Library/Developer/Toolchains/swift-latest/usr/bin:"${PATH}"

(fish の人用にも念のため)

set -x PATH /Library/Developer/Toolchains/swift-latest/usr/bin $PATH

(補足)Xcode からも使いたい場合は以下のリンクを参考に Toolchain を変更する

https://github.com/tensorflow/swift/blob/master/Installation.md#installation

swift-models のレポジトリを clone

以下のレポジトリを clone する https://github.com/tensorflow/swift-models

ちなみにこのレポジトリはcontributeできるので、今後他のモデルも追加されて行きそう

git clone git@github.com:tensorflow/swift-models.git

ビルド

上記の clone したディレクトリに行き、以下のコマンドを実行する

swift -O MNIST.swift

これで MNIST のビルドが始まる

ちなみに -O は compile with optimization

これでビルドができると思う