ユーザ用ツール

サイト用ツール


ai:tensorflow

差分

このページの2つのバージョン間の差分を表示します。

この比較画面へのリンク

両方とも前のリビジョン前のリビジョン
次のリビジョン
前のリビジョン
ai:tensorflow [2019/09/07 12:44] – [Python] ogaai:tensorflow [2019/09/07 19:38] (現在) – [Source からビルド] oga
行 19: 行 19:
 また TensorFlow 自体にも Keras が統合されています。 また TensorFlow 自体にも Keras が統合されています。
  
 +
 +  * default では画像フォーマットが他のフレームワークと異なり channels_last (NHWC) になっています。
 +    * channels_first (NCHW) の方が GPU 上では高速です。
 +    * CPU ではデフォルトでは channels_first (NCHW) が利用できません。channels_first に対応するには MKL を有効にしてビルドし直す必要があります。
 +
 +
 +
 +----
  
 ===== Windows ===== ===== Windows =====
行 43: 行 51:
  
 ==== Source からの build 手順 ==== ==== Source からの build 手順 ====
 +
 +  * cmake を使う方法と bazel を使う方法があります。公式サイトのビルド手順では bazel を使用しています。
 +
 +=== CMake の場合 (古い Version 時の情報です) ===
  
   * Windows による build memo   * Windows による build memo
行 68: 行 80:
       * Ryzen 7 1800X 4時間前後       * Ryzen 7 1800X 4時間前後
       * Core i7-6700K 6時間前後       * Core i7-6700K 6時間前後
- 
  
  
行 77: 行 88:
  
 ===== Linux ===== ===== Linux =====
 +
 +CUDA (GeForce) 及び ROCm (RADEON) が利用できます。
 +
 +
 +==== Python ====
  
 <code> <code>
 $ sudo apt install python3-pip python3-venv $ sudo apt install python3-pip python3-venv
 +$ pip3 install tensorflow-gpu
 </code> </code>
  
 +  * ROCm
 +    * [[https://rocm.github.io/tensorflow.html]]
 +
 +==== Source からビルド ====
 +
 +bazel を使います。
 +CPU または CUDA を使う場合は、公式サイト通りの手順でビルド可能です。
 +ROCm 対応版は専用のリポジトリから取得してください。
 +
 +  * C API 向けのビルドで 1時間くらいかかります。
 +  * <nowiki>C++</nowiki> API はヘッダの収集やライブラリのリンクが難しいので C API の方がおすすめです。
 +
 +とりあえず確認したもの
 +
 +  * 公式版 r1.14 CUDA C API, CPU + MKL C API
 +  * ROCm 版 r1.14-rocm C API
  
  
  
ai/tensorflow.1567827895.txt.gz · 最終更新: 2019/09/07 12:44 by oga

Donate Powered by PHP Valid HTML5 Valid CSS Driven by DokuWiki