Crie a partir do código-fonte no Windows

Compile um pacote pip do TensorFlow a partir do código-fonte e instale-o no Windows.

Configuração para Windows

Instale as seguintes ferramentas de compilação para configurar seu ambiente de desenvolvimento Windows.

Instale o Python e as dependências do pacote TensorFlow.

Instale o Python 3.9 ou superior de 64 bits para Windows . Selecione o pip como um recurso opcional e adicione-o à sua variável de ambiente %PATH% .

Instale as dependências do pacote TensorFlow via pip :

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

As dependências estão listadas no arquivo setup.py em REQUIRED_PACKAGES .

Instalar Bazel

Instale o Bazel , a ferramenta de compilação usada para compilar o TensorFlow. Para obter informações sobre a versão do Bazel, consulte as configurações de compilação testadas para Windows. Configure o Bazel para compilar em C++ .

Adicione o caminho do executável do Bazel à sua variável de ambiente %PATH% .

Instale o MSYS2

Instale o MSYS2 para obter as ferramentas binárias necessárias para compilar o TensorFlow. Se o MSYS2 estiver instalado em C:\msys64 , adicione C:\msys64\usr\bin à sua variável de ambiente %PATH% . Em seguida, usando cmd.exe , execute:

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

Instalar as Ferramentas de Compilação do Visual C++ 2022

Instale as ferramentas de compilação do Visual C++ 2022. Elas vêm com o Visual Studio Community 2022 , mas podem ser instaladas separadamente:

1. Acesse a página de downloads do Visual Studio .
2. Selecione Ferramentas para Visual Studio ou Outras Ferramentas, Frameworks e Redistribuíveis .
3. Baixe e instale:
   • Ferramentas de compilação para Visual Studio 2022
   • Pacotes redistribuíveis do Microsoft Visual C++ para Visual Studio 2022

Instale o LLVM

1. Acesse a página de downloads do LLVM .
2. Baixe e instale o LLVM compatível com Windows em C:/Program Files/LLVM, por exemplo, LLVM-17.0.6-win64.exe.

Instalar suporte para GPU (opcional)

Consulte o guia de suporte a GPUs do Windows para instalar os drivers e o software adicional necessário para executar o TensorFlow em uma GPU.

Baixe o código-fonte do TensorFlow.

Utilize o Git para clonar o repositório do TensorFlow ( git já está instalado com o MSYS2):

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

O repositório usa por padrão a branch de desenvolvimento master . Você também pode selecionar uma branch de lançamento para compilar:

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

Opcional: Configuração de variáveis ​​de ambiente

Execute os seguintes comandos antes de executar o comando de compilação para evitar problemas com a criação do pacote: (Se os comandos abaixo foram configurados durante a instalação dos pacotes, ignore-os). Execute set para verificar se todos os caminhos foram configurados corretamente e execute echo %Environmental Variable% , por exemplo, echo %BAZEL_VC% para verificar o caminho configurado para uma variável de ambiente específica.

Problema de configuração do caminho do 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 

Problema de configuração do caminho 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)

Opcional: Configure a compilação

As compilações do TensorFlow são configuradas pelo arquivo .bazelrc no diretório raiz do repositório. Os scripts ./configure ou ./configure.py podem ser usados ​​para ajustar configurações comuns.

Se precisar alterar a configuração, execute o script ./configure a partir do diretório raiz do repositório.

python ./configure.py

Este script solicita a localização das dependências do TensorFlow e pede opções adicionais de configuração de compilação (como flags do compilador). A seguir, um exemplo de execução do python ./configure.py (sua sessão pode ser diferente):

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.

Compile e instale o pacote pip.

O pacote pip é construído em duas etapas. Um comando bazel build cria um programa "package-builder". Em seguida, você executa o package-builder para criar o pacote.

Construa o construtor de pacotes

O repositório tensorflow:master foi atualizado para compilar a versão 2.x por padrão. Instale o Bazel e use bazel build para criar o pacote do TensorFlow.

bazel build //tensorflow/tools/pip_package:wheel

somente CPU

Use bazel para criar o pacote TensorFlow com suporte apenas para CPU:

Compile com MSVC

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

Construa com CLANG

Use --config= win_clang para compilar o TenorFlow com o compilador CLANG:

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

suporte a GPU

Para criar o construtor de pacotes do TensorFlow com suporte a GPU:

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

Comandos para limpar o cache do Bazel e resolver erros causados ​​por dados em cache inválidos ou desatualizados. O comando `bazel clean` com a flag `--expunge` remove arquivos permanentemente.

bazel clean 
bazel clean --expunge  

Opções de compilação do Bazel

Use esta opção ao compilar para evitar problemas com a criação do pacote: tensorflow:issue#22390

--define=no_tensorflow_py_deps=true

Consulte a referência da linha de comando do Bazel para obter opções de compilação .

Compilar o TensorFlow a partir do código-fonte pode usar muita RAM. Se o seu sistema tiver memória limitada, limite o uso de RAM do Bazel com: --local_ram_resources=2048 .

Se estiver compilando com suporte a GPU, adicione --copt=-nvcc_options=disable-warnings para suprimir as mensagens de aviso do nvcc.

Construa o pacote

Para criar um pacote pip, você precisa especificar a flag `--repo_env=WHEEL_NAME`. Dependendo do nome fornecido, o pacote será criado. Por exemplo:

Para compilar o pacote TensorFlow para CPU:

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

Para criar o pacote nightly, defina tf_nightly em vez de tensorflow , por exemplo, para criar o pacote nightly para CPU:

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

Como resultado, a roda gerada estará localizada em

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

Instale o pacote

O nome do arquivo .whl gerado depende da versão do TensorFlow e da sua plataforma. Use pip install para instalar o pacote, por exemplo:

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

Compile usando o shell MSYS

O TensorFlow também pode ser compilado usando o shell MSYS. Faça as alterações listadas abaixo e, em seguida, siga as instruções anteriores para a linha de comando nativa do Windows ( cmd.exe ).

Desativar conversão de caminho MSYS

O MSYS converte automaticamente argumentos que se parecem com caminhos Unix em caminhos do Windows, e isso não funciona com bazel . (O rótulo //path/to:bin é considerado um caminho absoluto Unix, pois começa com uma barra.)

export MSYS_NO_PATHCONV=1
export MSYS2_ARG_CONV_EXCL="*"

Defina seu CAMINHO

Adicione os diretórios de instalação do Bazel e do Python à sua variável de ambiente $PATH . Se o Bazel estiver instalado em C:\tools\bazel.exe e o Python em C:\Python\python.exe , defina seu PATH com:

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

Para suporte a GPU, adicione os diretórios binários do CUDA e do cuDNN ao seu $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"

Configurações de compilação testadas

CPU

GPU