TensorFlow.jsコンバータ(tensorflowjs)のインストール [Docker/nvidia-docker2]

Python/TensorFlowの使い方(目次)

「TensorFlow.jsコンバータ」(tensorflowjs/tfjs-converter)はTensorFlowの「SavedModel」「Keras h5」「Session Bundle」「Frozen Model」「Tensorflow Hub module」形式のモデルをTensorFlow.jsで使用できる「web-friendly format」に変換するツールです。

注意事項

TensorFlow.jsコンバータをインストールすると

も一緒にインストールされます。TensorFlow(GPU版)を使用している方は「GPU版が無効」となりCPUが有効となります。

それを回避する為には別コンテナにコンバータをインストールして下さい。

コンテナの作成

tensorflowイメージの「tensorjs」というコンテナを作成します。

docker run --name tensorjs -d -it -p 7777:7777 -v /home/ユーザー名/tensor:/foo tensorflow/tensorflow bash

※Dockerについてはこちらをご参考にしてください。

インストール

コンテナ上で実行します。

pip install tensorflowjs

参考サイト

TensorFlow.jsコンバータ (公式)

• Twitter
• Facebook
• はてブ

• Pocket
• LINE

関連記事