Windows 上のソースからビルドする,Windows 上のソースからビルドする

ソースから TensorFlow pipパッケージをビルドし、Windows にインストールします。

Windowsのセットアップ

Windows 開発環境を構成するには、次のビルド ツールをインストールします。

PythonとTensorFlowパッケージの依存関係をインストールする

Windows 用の Python 3.9+ 64 ビットリリースをインストールします。オプション機能としてpipを選択し、 %PATH%環境変数に追加します。

TensorFlow pipパッケージの依存関係をインストールします。

pip3 install -U pip
pip3 install -U six numpy wheel packaging
pip3 install -U keras_preprocessing --no-deps

依存関係は、 setup.pyファイルのREQUIRED_PACKAGESの下にリストされています。

Bazelをインストールする

TensorFlow をコンパイルするためのビルドツールであるBazel をインストールします。Bazel のバージョンについては、Windows でテスト済みのビルド構成を参照してください。C ++ をビルドするように Bazel を設定します。

Bazel 実行可能ファイルの場所を%PATH%環境変数に追加します。

MSYS2をインストールする

TensorFlowのビルドに必要なbinツールとしてMSYS2をインストールしてください。MSYS2がC:\msys64にインストールされている場合は、 C:\msys64\usr\bin %PATH%環境変数に追加してください。その後、 cmd.exeを使用して以下を実行します。

pacman -Syu (requires a console restart)
pacman -S git patch unzip
pacman -S git patch unzip rsync

Visual C++ ビルドツール 2022 をインストールする

Visual C++ ビルドツール 2022をインストールします。これはVisual Studio Community 2022に付属していますが、別途インストールすることもできます。

1. Visual Studioのダウンロードにアクセスし、
2. Visual Studioのツールまたはその他のツール、フレームワークと再頒布可能パッケージを選択し、
3. ダウンロードしてインストール:
   • Visual Studio 2022 のビルド ツール
   • Visual Studio 2022 用 Microsoft Visual C++ 再頒布可能パッケージ

LLVMをインストールする

1. LLVMのダウンロードへアク​​セスし、
2. Windows 互換の LLVM を C:/Program Files/LLVM にダウンロードしてインストールします (例: LLVM-17.0.6-win64.exe)

GPUサポートをインストールする（オプション）

GPU で TensorFlow を実行するために必要なドライバーと追加ソフトウェアをインストールするには、Windows GPU サポートガイドを参照してください。

TensorFlowのソースコードをダウンロードする

Gitを使用してTensorFlow リポジトリのクローンを作成します ( gitは MSYS2 とともにインストールされます)。

git clone https://github.com/tensorflow/tensorflow.git
cd tensorflow

リポジトリはデフォルトでmaster開発ブランチに設定されています。リリースブランチをチェックアウトしてビルドすることもできます。

git checkout branch_name  # r1.9, r1.10, etc.

オプション: 環境変数の設定

パッケージ作成に関する問題を回避するため、ビルドコマンドを実行する前に以下のコマンドを実行してください。(パッケージのインストール時に以下のコマンドが設定されている場合は無視してください。) setを実行してすべてのパスが正しく設定されているかどうかを確認します。echo echo %Environmental Variable% (例: echo %BAZEL_VC%を実行して特定の環境変数のパス設定を確認します。

Python パス設定の問題tensorflow:issue#59943 、 tensorflow:issue#9436 、 tensorflow:issue#60083

set PATH=path/to/python;%PATH% # [e.g. (C:/Python311)]
set PATH=path/to/python/Scripts;%PATH% # [e.g. (C:/Python311/Scripts)] 
set PYTHON_BIN_PATH=path/to/python_virtualenv/Scripts/python.exe 
set PYTHON_LIB_PATH=path/to/python virtualenv/lib/site-packages 
set PYTHON_DIRECTORY=path/to/python_virtualenv/Scripts 

Bazel/MSVC/CLANG パス設定の問題tensorflow:issue#54578

set BAZEL_SH=C:/msys64/usr/bin/bash.exe 
set BAZEL_VS=C:/Program Files/Microsoft Visual Studio/2022/BuildTools 
set BAZEL_VC=C:/Program Files/Microsoft Visual Studio/2022/BuildTools/VC 
set Bazel_LLVM=C:/Program Files/LLVM (explicitly tell Bazel where LLVM is installed by BAZEL_LLVM, needed while using CLANG)
set PATH=C:/Program Files/LLVM/bin;%PATH% (Optional, needed while using CLANG as Compiler)

オプション: ビルドを構成する

TensorFlow ビルドは./configureリポジトリのルートディレクトリにある.bazelrcファイルによって構成されます。./configure または./configure.pyスクリプトを使用して、一般的な設定を調整できます。

設定を変更する必要がある場合は、リポジトリのルート ディレクトリから./configureスクリプトを実行します。

python ./configure.py

このスクリプトは、TensorFlowの依存関係の場所と、追加のビルド設定オプション（コンパイラフラグなど）の入力を求めます。以下はpython ./configure.pyの実行例です（セッションによって異なる場合があります）。

python ./configure.py
You have bazel 6.5.0 installed.
Please specify the location of python. [Default is C:\Python311\python.exe]:

Found possible Python library paths:
C:\Python311\lib\site-packages
Please input the desired Python library path to use.  Default is [C:\Python311\lib\site-packages]

Do you wish to build TensorFlow with ROCm support? [y/N]:
No ROCm support will be enabled for TensorFlow.

Warning: Cannot build with CUDA support on Windows.
Starting in TF 2.11, CUDA build is not supported for Windows. To use TensorFlow GPU on Windows, you will need to build/install TensorFlow in WSL2.

Do you want to use Clang to build TensorFlow? [Y/n]:
Add "--config=win_clang" to compile TensorFlow with CLANG.

Please specify the path to clang executable. [Default is C:\Program Files\LLVM\bin\clang.EXE]:

You have Clang 17.0.6 installed.

Please specify optimization flags to use during compilation when bazel option "--config=opt" is specified [Default is /arch:AVX]:

Would you like to override eigen strong inline for some C++ compilation to reduce the compilation time? [Y/n]:
Eigen strong inline overridden.

Would you like to interactively configure ./WORKSPACE for Android builds? [y/N]:
Not configuring the WORKSPACE for Android builds.

Preconfigured Bazel build configs. You can use any of the below by adding "--config=<>" to your build command. See .bazelrc for more details.
        --config=mkl            # Build with MKL support.
        --config=mkl_aarch64    # Build with oneDNN and Compute Library for the Arm Architecture (ACL).
        --config=monolithic     # Config for mostly static monolithic build.
        --config=numa           # Build with NUMA support.
        --config=dynamic_kernels        # (Experimental) Build kernels into separate shared objects.
        --config=v1             # Build with TensorFlow 1 API instead of TF 2 API.
Preconfigured Bazel build configs to DISABLE default on features:
        --config=nogcp          # Disable GCP support.
        --config=nonccl         # Disable NVIDIA NCCL support.

pipパッケージをビルドしてインストールする

pip パッケージは2段階でビルドされます。bazel bazel buildコマンドは「package-builder」プログラムを作成します。その後、package-builder を実行してパッケージを作成します。

パッケージビルダーをビルドする

tensorflow:master リポジトリが更新され、デフォルトで 2.x がビルドされるようになりました。Bazelをインストールし、 bazel buildを使用して TensorFlow パッケージビルダーを作成してください。

bazel build //tensorflow/tools/pip_package:wheel

CPUのみ

bazelを使用して、CPU のみをサポートする TensorFlow パッケージ ビルダーを作成します。

MSVCでビルドする

bazel build --config=opt --repo_env=TF_PYTHON_VERSION=3.11 //tensorflow/tools/pip_package:wheel --repo_env=WHEEL_NAME=tensorflow_cpu

CLANGでビルドする

CLANG コンパイラを使用して TenorFlow をビルドするには、--config= win_clangを使用します。

bazel build --config=win_clang --repo_env=TF_PYTHON_VERSION=3.11 //tensorflow/tools/pip_package:wheel --repo_env=WHEEL_NAME=tensorflow_cpu

GPUサポート

GPU サポート付きの TensorFlow パッケージ ビルダーを作成するには:

bazel build --config=opt --config=cuda --define=no_tensorflow_py_deps=true //tensorflow/tools/pip_package:build_pip_package

無効または古いキャッシュデータによるエラーを解決するために、bazel キャッシュを消去するコマンド。--expunge フラグを指定した bazel clean はファイルを永久に削除します。

bazel clean 
bazel clean --expunge  

Bazelビルドオプション

パッケージ作成に関する問題を回避するには、ビルド時にこのオプションを使用します: tensorflow:issue#22390

--define=no_tensorflow_py_deps=true

ビルド オプションについては、Bazelコマンドライン リファレンスを参照してください。

TensorFlowをソースからビルドすると、大量のRAMが消費される可能性があります。システムのメモリ容量が限られている場合は、 --local_ram_resources=2048を指定してBazelのRAM使用量を制限してください。

GPU サポートを使用してビルドする場合は、 --copt=-nvcc_options=disable-warningsを追加して、nvcc 警告メッセージを抑制します。

パッケージをビルドする

pip パッケージをビルドするには、--repo_env=WHEEL_NAME フラグを指定する必要があります。指定された名前に応じて、パッケージが作成されます。例:

Tensorflow CPU パッケージをビルドするには:

bazel build //tensorflow/tools/pip_package:wheel --repo_env=WHEEL_NAME=tensorflow_cpu

ナイトリー パッケージをビルドするには、 tensorflowの代わりにtf_nightly設定します。たとえば、CPU ナイトリー パッケージをビルドするには、次のようにします。

bazel build //tensorflow/tools/pip_package:wheel --repo_env=WHEEL_NAME=tf_nightly_cpu

その結果、生成されたホイールは

bazel-bin/tensorflow/tools/pip_package/wheel_house/

パッケージをインストールする

生成される.whlファイルのファイル名は、TensorFlowのバージョンとプラットフォームによって異なります。パッケージをインストールするには、 pip install使用してください。例：

pip install bazel-bin/tensorflow/tools/pip_package/wheel_house/tensorflow-version-tags.whl

MSYSシェルを使用してビルドする

TensorFlowはMSYSシェルを使ってビルドすることもできます。以下の変更を加えた後、Windowsネイティブのコマンドライン（ cmd.exe ）の手順に従ってください。

MSYSパス変換を無効にする

MSYS は Unix パスのように見える引数を Windows パスに自動的に変換しますが、これはbazelでは機能しません。(ラベル//path/to:binスラッシュで始まるため、Unix の絶対パスとみなされます。)

export MSYS_NO_PATHCONV=1
export MSYS2_ARG_CONV_EXCL="*"

PATHを設定する

BazelとPythonのインストールディレクトリを$PATH環境変数に追加します。BazelがC:\tools\bazel.exeに、PythonがC:\Python\python.exeにインストールされている場合は、 PATHを次のように設定します。

# Use Unix-style with ':' as separator
export PATH="/c/tools:$PATH"
export PATH="/c/path/to/Python:$PATH"

GPU サポートの場合は、CUDA および cuDNN bin ディレクトリを$PATHに追加します。

export PATH="/c/Program Files/NVIDIA GPU Computing Toolkit/CUDA/v11.0/bin:$PATH"
export PATH="/c/Program Files/NVIDIA GPU Computing Toolkit/CUDA/v11.0/extras/CUPTI/libx64:$PATH"
export PATH="/c/tools/cuda/bin:$PATH"

テスト済みのビルド構成

CPU

グラフィックプロセッサ