ai:tensorflow
差分
このページの2つのバージョン間の差分を表示します。
両方とも前のリビジョン前のリビジョン次のリビジョン | 前のリビジョン最新のリビジョン両方とも次のリビジョン | ||
ai:tensorflow [2019/09/07 12:46] – [Source からの build 手順] oga | ai:tensorflow [2019/09/07 13:56] – [Python] oga | ||
---|---|---|---|
行 19: | 行 19: | ||
また TensorFlow 自体にも Keras が統合されています。 | また TensorFlow 自体にも Keras が統合されています。 | ||
+ | |||
+ | * default では画像フォーマットが他のフレームワークと異なり channels_last (NHWC) になっています。 | ||
+ | * channels_first (NCHW) の方が GPU 上では高速です。 | ||
+ | * CPU ではデフォルトでは channels_first (NCHW) が利用できません。channels_first に対応するには MKL を有効にしてビルドし直す必要があります。 | ||
+ | |||
+ | |||
+ | |||
+ | ---- | ||
===== Windows ===== | ===== Windows ===== | ||
行 44: | 行 52: | ||
==== Source からの build 手順 ==== | ==== Source からの build 手順 ==== | ||
- | * cmake を使う方法と bazel を使う方法があります。公式サイトでは bazel を使用しています。 | + | * cmake を使う方法と bazel を使う方法があります。公式サイトのビルド手順では bazel を使用しています。 |
- | === CMake の場合 === | + | === CMake の場合 |
* Windows による build memo | * Windows による build memo | ||
行 80: | 行 88: | ||
===== Linux ===== | ===== Linux ===== | ||
+ | |||
+ | CUDA (GeForce) 及び ROCm (RADEON) が利用できます。 | ||
+ | |||
+ | |||
+ | ==== Python ==== | ||
< | < | ||
$ sudo apt install python3-pip python3-venv | $ sudo apt install python3-pip python3-venv | ||
+ | $ pip3 install tensorflow-gpu | ||
</ | </ | ||
+ | * ROCm | ||
+ | * [[https:// | ||
+ | |||
+ | ==== Source からビルド ==== | ||
+ | |||
+ | bazel を使います。 | ||
+ | CPU または CUDA を使う場合は、公式サイト通りの手順でビルド可能です。 | ||
+ | ROCm 対応版は専用のリポジトリから取得します。 | ||
+ | * C API 向けのビルドで 1時間くらいかかります。 | ||
+ | * < | ||
ai/tensorflow.txt · 最終更新: 2019/09/07 19:38 by oga