Compilar a partir da fonte no Windows

Crie 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 do Windows.

Instale o Python e as dependências do pacote TensorFlow

Instale uma versão do Python 3.9+ de 64 bits para Windows . Selecione pip como um recurso opcional e adicione-o à sua variável de ambiente %PATH% .

Instale as dependências do pacote pip do TensorFlow:

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 o Bazel

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

Adicione a localização do executável do Bazel à sua variável de ambiente %PATH% .

Instalar MSYS2

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

pacman -S git patch unzip

Instale o Visual C++ Build Tools 2019

Instale as ferramentas de compilação do Visual C++ 2019 . Isso vem com o Visual Studio 2019 , mas pode ser instalado separadamente:

  1. Vá para os downloads do Visual Studio ,
  2. Selecione Redistribuíveis e Ferramentas de Construção ,
  3. Baixar e instalar:
    • Microsoft Visual C++ 2019 Redistribuível
    • Ferramentas de compilação da Microsoft 2019

Instalar suporte a GPU (opcional)

Consulte o guia de suporte de GPU 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

Use o Git para clonar o repositório do TensorFlow ( git é instalado com o MSYS2):

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

O repositório é padronizado para a ramificação de desenvolvimento master . Você também pode verificar uma ramificação de lançamento para construir:

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

Opcional: configuração de variável ambiental

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

Problema de configuração do caminho do Python tensorflow:issue#59943 , tensorflow:issue#9436 , tensorflow:issue#60083

set PATH=path/to/python # [e.g. (C:/Python310)]
set PATH=path/to/python/Scripts # [e.g. (C:/Python310/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 de caminho Bazel/MSVC tensorflow:issue#54578

set BAZEL_SH=C:/msys64/usr/bin/bash.exe 
set BAZEL_VS=C:/Program Files(x86)/Microsoft Visual Studio/2019/BuildTools 
set BAZEL_VC=C:/Program Files(x86)/Microsoft Visual Studio/2019/BuildTools/VC 

Opcional: configurar 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 no diretório raiz do repositório.

python ./configure.py

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

Crie e instale o pacote pip

O pacote pip é construído em duas etapas. Um comando bazel build cria um programa de "construtor de pacotes". Em seguida, execute o construtor de pacotes para criar o pacote.

Crie o construtor de pacotes

tensorflow:master repo foi atualizado para construir 2.x por padrão. Instale o Bazel e use bazel build para criar o construtor de pacotes TensorFlow.

bazel build //tensorflow/tools/pip_package:build_pip_package

somente CPU

Use bazel para criar o construtor de pacotes TensorFlow com suporte somente para CPU:

bazel build --config=opt //tensorflow/tools/pip_package:build_pip_package

suporte GPU

Para criar o construtor de pacotes 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 para resolver erros devido a dados armazenados em cache inválidos ou desatualizados, bazel clean com --expunge sinalizador 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 de linha de comando Bazel para opções de compilação .

Construir o TensorFlow a partir da fonte pode usar muita RAM. Se o seu sistema tiver restrição de memória, limite o uso de RAM do Bazel com: --local_ram_resources=2048 .

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

Crie o pacote

O comando bazel build cria um executável chamado build_pip_package — este é o programa que cria o pacote pip . Por exemplo, o seguinte cria um pacote .whl no diretório C:/tmp/tensorflow_pkg :

bazel-bin\tensorflow\tools\pip_package\build_pip_package C:/tmp/tensorflow_pkg

Embora seja possível criar configurações CUDA e não CUDA na mesma árvore de origem, recomendamos executar bazel clean ao alternar entre essas duas configurações na mesma árvore de origem.

Instale o pacote

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

pip3 install C:/tmp/tensorflow_pkg/tensorflow-version-tags.whl

e.g. pip3 install C:/tmp/tensorflow_pkg/tensorflow-2.12.0-cp310-cp310-win_amd64.whl

Construir usando o shell MSYS

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

Desativar conversão de caminho MSYS

MSYS converte automaticamente argumentos que se parecem com caminhos Unix em caminhos Windows, e isso não funciona com bazel . (O rótulo //path/to:bin é considerado um caminho absoluto do 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 Bazel e 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 de GPU, adicione os diretórios bin CUDA e 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

Versão versão Python Compilador ferramentas de construção
tensorflow-2.12.0 3.8-3.11 MSVC 2019 Bazel 5.3.0
tensorflow-2.11.0 3.7-3.10 MSVC 2019 Bazel 5.3.0
tensorflow-2.10.0 3.7-3.10 MSVC 2019 Bazel 5.1.1
tensorflow-2.9.0 3.7-3.10 MSVC 2019 Bazel 5.0.0
tensorflow-2.8.0 3.7-3.10 MSVC 2019 Bazel 4.2.1
tensorflow-2.7.0 3.7-3.9 MSVC 2019 Bazel 3.7.2
tensorflow-2.6.0 3.6-3.9 MSVC 2019 Bazel 3.7.2
tensorflow-2.5.0 3.6-3.9 MSVC 2019 Bazel 3.7.2
tensorflow-2.4.0 3,6-3,8 MSVC 2019 Bazel 3.1.0
tensorflow-2.3.0 3,5-3,8 MSVC 2019 Bazel 3.1.0
tensorflow-2.2.0 3,5-3,8 MSVC 2019 Bazel 2.0.0
tensorflow-2.1.0 3,5-3,7 MSVC 2019 Bazel 0.27.1-0.29.1
tensorflow-2.0.0 3,5-3,7 MSVC 2017 Bazel 0.26.1
tensorflow-1.15.0 3,5-3,7 MSVC 2017 Bazel 0.26.1
tensorflow-1.14.0 3,5-3,7 MSVC 2017 Bazel 0.24.1-0.25.2
tensorflow-1.13.0 3,5-3,7 MSVC 2015 atualização 3 Bazel 0.19.0-0.21.0
tensorflow-1.12.0 3,5-3,6 MSVC 2015 atualização 3 Bazel 0.15.0
tensorflow-1.11.0 3,5-3,6 MSVC 2015 atualização 3 Bazel 0.15.0
tensorflow-1.10.0 3,5-3,6 MSVC 2015 atualização 3 Cmake v3.6.3
tensorflow-1.9.0 3,5-3,6 MSVC 2015 atualização 3 Cmake v3.6.3
tensorflow-1.8.0 3,5-3,6 MSVC 2015 atualização 3 Cmake v3.6.3
tensorflow-1.7.0 3,5-3,6 MSVC 2015 atualização 3 Cmake v3.6.3
tensorflow-1.6.0 3,5-3,6 MSVC 2015 atualização 3 Cmake v3.6.3
tensorflow-1.5.0 3,5-3,6 MSVC 2015 atualização 3 Cmake v3.6.3
tensorflow-1.4.0 3,5-3,6 MSVC 2015 atualização 3 Cmake v3.6.3
tensorflow-1.3.0 3,5-3,6 MSVC 2015 atualização 3 Cmake v3.6.3
tensorflow-1.2.0 3,5-3,6 MSVC 2015 atualização 3 Cmake v3.6.3
tensorflow-1.1.0 3.5 MSVC 2015 atualização 3 Cmake v3.6.3
tensorflow-1.0.0 3.5 MSVC 2015 atualização 3 Cmake v3.6.3

GPU

Versão versão Python Compilador ferramentas de construção cuDNN CUDA
tensorflow_gpu-2.10.0 3.7-3.10 MSVC 2019 Bazel 5.1.1 8.1 11.2
tensorflow_gpu-2.9.0 3.7-3.10 MSVC 2019 Bazel 5.0.0 8.1 11.2
tensorflow_gpu-2.8.0 3.7-3.10 MSVC 2019 Bazel 4.2.1 8.1 11.2
tensorflow_gpu-2.7.0 3.7-3.9 MSVC 2019 Bazel 3.7.2 8.1 11.2
tensorflow_gpu-2.6.0 3.6-3.9 MSVC 2019 Bazel 3.7.2 8.1 11.2
tensorflow_gpu-2.5.0 3.6-3.9 MSVC 2019 Bazel 3.7.2 8.1 11.2
tensorflow_gpu-2.4.0 3,6-3,8 MSVC 2019 Bazel 3.1.0 8,0 11,0
tensorflow_gpu-2.3.0 3,5-3,8 MSVC 2019 Bazel 3.1.0 7.6 10.1
tensorflow_gpu-2.2.0 3,5-3,8 MSVC 2019 Bazel 2.0.0 7.6 10.1
tensorflow_gpu-2.1.0 3,5-3,7 MSVC 2019 Bazel 0.27.1-0.29.1 7.6 10.1
tensorflow_gpu-2.0.0 3,5-3,7 MSVC 2017 Bazel 0.26.1 7.4 10
tensorflow_gpu-1.15.0 3,5-3,7 MSVC 2017 Bazel 0.26.1 7.4 10
tensorflow_gpu-1.14.0 3,5-3,7 MSVC 2017 Bazel 0.24.1-0.25.2 7.4 10
tensorflow_gpu-1.13.0 3,5-3,7 MSVC 2015 atualização 3 Bazel 0.19.0-0.21.0 7.4 10
tensorflow_gpu-1.12.0 3,5-3,6 MSVC 2015 atualização 3 Bazel 0.15.0 7.2 9,0
tensorflow_gpu-1.11.0 3,5-3,6 MSVC 2015 atualização 3 Bazel 0.15.0 7 9
tensorflow_gpu-1.10.0 3,5-3,6 MSVC 2015 atualização 3 Cmake v3.6.3 7 9
tensorflow_gpu-1.9.0 3,5-3,6 MSVC 2015 atualização 3 Cmake v3.6.3 7 9
tensorflow_gpu-1.8.0 3,5-3,6 MSVC 2015 atualização 3 Cmake v3.6.3 7 9
tensorflow_gpu-1.7.0 3,5-3,6 MSVC 2015 atualização 3 Cmake v3.6.3 7 9
tensorflow_gpu-1.6.0 3,5-3,6 MSVC 2015 atualização 3 Cmake v3.6.3 7 9
tensorflow_gpu-1.5.0 3,5-3,6 MSVC 2015 atualização 3 Cmake v3.6.3 7 9
tensorflow_gpu-1.4.0 3,5-3,6 MSVC 2015 atualização 3 Cmake v3.6.3 6 8
tensorflow_gpu-1.3.0 3,5-3,6 MSVC 2015 atualização 3 Cmake v3.6.3 6 8
tensorflow_gpu-1.2.0 3,5-3,6 MSVC 2015 atualização 3 Cmake v3.6.3 5.1 8
tensorflow_gpu-1.1.0 3.5 MSVC 2015 atualização 3 Cmake v3.6.3 5.1 8
tensorflow_gpu-1.0.0 3.5 MSVC 2015 atualização 3 Cmake v3.6.3 5.1 8