Instale o TensorFlow com pip

Linux

python3 -m pip install 'tensorflow[and-cuda]'
# Verify the installation:
python3 -c "import tensorflow as tf; print(tf.config.list_physical_devices('GPU'))"

MacOS

# There is currently no official GPU support for MacOS.
python3 -m pip install tensorflow
# Verify the installation:
python3 -c "import tensorflow as tf; print(tf.reduce_sum(tf.random.normal([1000, 1000])))"

Nativo do Windows

conda install -c conda-forge cudatoolkit=11.2 cudnn=8.1.0
# Anything above 2.10 is not supported on the GPU on Windows Native
python -m pip install "tensorflow<2.11"
# Verify the installation:
python -c "import tensorflow as tf; print(tf.config.list_physical_devices('GPU'))"

Windows WSL2

python3 -m pip install tensorflow[and-cuda]
# Verify the installation:
python3 -c "import tensorflow as tf; print(tf.config.list_physical_devices('GPU'))"

CPU

python3 -m pip install tensorflow
# Verify the installation:
python3 -c "import tensorflow as tf; print(tf.reduce_sum(tf.random.normal([1000, 1000])))"

Noturno

python3 -m pip install tf-nightly
# Verify the installation:
python3 -c "import tensorflow as tf; print(tf.reduce_sum(tf.random.normal([1000, 1000])))"

Linux

1. Requisitos do sistema

• Ubuntu 16.04 ou superior (64 bits)

O TensorFlow oferece suporte oficial apenas para Ubuntu. No entanto, as instruções a seguir também podem funcionar em outras distribuições Linux.

2. Configuração da GPU

Você pode pular esta seção se executar o TensorFlow apenas na CPU.

Instale o driver da GPU NVIDIA, caso ainda não o tenha feito. Você pode usar o seguinte comando para verificar se ele está instalado.

nvidia-smi

3. Crie um ambiente virtual com o venv.

O módulo venv faz parte da biblioteca padrão do Python e é a maneira oficialmente recomendada de criar ambientes virtuais.

Navegue até o diretório de ambientes virtuais desejado e crie um novo ambiente venv chamado tf com o seguinte comando.

python3 -m venv tf 

Você pode ativá-lo com o seguinte comando.

source tf/bin/activate    

Certifique-se de que o ambiente virtual esteja ativado para o restante da instalação.

4. Instale o TensorFlow

O TensorFlow requer uma versão recente do pip, portanto, atualize sua instalação do pip para garantir que você esteja executando a versão mais recente.

pip install --upgrade pip

Em seguida, instale o TensorFlow com o pip.

# For GPU users
pip install tensorflow[and-cuda]
# For CPU users
pip install tensorflow

6. Verifique a instalação

Verifique a configuração da CPU:

python3 -c "import tensorflow as tf; print(tf.reduce_sum(tf.random.normal([1000, 1000])))"

Se um tensor for retornado, significa que você instalou o TensorFlow com sucesso.

Verifique a configuração da GPU:

python3 -c "import tensorflow as tf; print(tf.config.list_physical_devices('GPU'))"

Se uma lista de dispositivos de GPU for exibida, você instalou o TensorFlow com sucesso. Caso contrário, continue para a próxima etapa .

6. [Somente GPU] Configuração do ambiente virtual

Se o teste de GPU na seção anterior não foi bem-sucedido, a causa mais provável é que os componentes não estão sendo detectados e/ou estão em conflito com a instalação CUDA existente no sistema. Portanto, você precisa adicionar alguns links simbólicos para corrigir isso.

• Criar links simbólicos para bibliotecas compartilhadas da NVIDIA:

pushd $(dirname $(python -c 'print(__import__("tensorflow").__file__)'))
ln -svf ../nvidia/*/lib/*.so* .
popd

• Crie um link simbólico para ptxas:

ln -sf $(find $(dirname $(dirname $(python -c "import nvidia.cuda_nvcc;         
print(nvidia.cuda_nvcc.__file__)"))/*/bin/) -name ptxas -print -quit) $VIRTUAL_ENV/bin/ptxas

Verifique a configuração da GPU:

python3 -c "import tensorflow as tf; print(tf.config.list_physical_devices('GPU'))"

MacOS

1. Requisitos do sistema

• macOS 10.12.6 (Sierra) ou superior (64 bits)

Atualmente, não há suporte oficial para GPU na execução do TensorFlow no macOS. As instruções a seguir são para execução na CPU.

2. Verifique a versão do Python

Verifique se seu ambiente Python já está configurado:

python3 --version
python3 -m pip --version

3. Instale o TensorFlow

O TensorFlow requer uma versão recente do pip, portanto, atualize sua instalação do pip para garantir que você esteja executando a versão mais recente.

pip install --upgrade pip

Em seguida, instale o TensorFlow com o pip.

pip install tensorflow

4. Verifique a instalação

python3 -c "import tensorflow as tf; print(tf.reduce_sum(tf.random.normal([1000, 1000])))"

Se um tensor for retornado, significa que você instalou o TensorFlow com sucesso.

Nativo do Windows

1. Requisitos do sistema

• Windows 7 ou superior (64 bits)

2. Instale o Microsoft Visual C++ Redistributable.

Instale o Microsoft Visual C++ Redistributable para Visual Studio 2015, 2017 e 2019. A partir da versão TensorFlow 2.1.0, o arquivo msvcp140_1.dll é necessário e está incluído neste pacote (e pode não estar presente em versões anteriores do pacote redistribuível). O pacote redistribuível já vem com o Visual Studio 2019 , mas também pode ser instalado separadamente.

1. Acesse a página de downloads do Microsoft Visual C++ .
2. Role a página para baixo até a seção Visual Studio 2015, 2017 e 2019 .
3. Baixe e instale o Microsoft Visual C++ Redistributable para Visual Studio 2015, 2017 e 2019 para a sua plataforma.

Certifique-se de que os caminhos longos estejam habilitados no Windows.

3. Instale o Miniconda

O Miniconda é a abordagem recomendada para instalar o TensorFlow com suporte a GPU. Ele cria um ambiente separado para evitar a alteração de qualquer software instalado no seu sistema. Esta também é a maneira mais fácil de instalar o software necessário, especialmente para a configuração com GPU.

Baixe o instalador do Miniconda para Windows . Clique duas vezes no arquivo baixado e siga as instruções na tela.

4. Crie um ambiente conda

Crie um novo ambiente conda chamado tf com o seguinte comando.

conda create --name tf python=3.9

Você pode desativá-lo e ativá-lo com os seguintes comandos.

conda deactivate
conda activate tf

Certifique-se de que esteja ativado para o restante da instalação.

5. Configuração da GPU

Você pode pular esta seção se estiver executando o TensorFlow apenas na CPU.

Primeiro instale o driver da placa de vídeo NVIDIA, caso ainda não o tenha feito.

Em seguida, instale o CUDA e o cuDNN com o conda.

conda install -c conda-forge cudatoolkit=11.2 cudnn=8.1.0

6. Instale o TensorFlow

O TensorFlow requer uma versão recente do pip, portanto, atualize sua instalação do pip para garantir que você esteja executando a versão mais recente.

pip install --upgrade pip

Em seguida, instale o TensorFlow com o pip.

# Anything above 2.10 is not supported on the GPU on Windows Native
pip install "tensorflow<2.11" 

7. Verifique a instalação

Verifique a configuração da CPU:

python -c "import tensorflow as tf; print(tf.reduce_sum(tf.random.normal([1000, 1000])))"

Se um tensor for retornado, significa que você instalou o TensorFlow com sucesso.

Verifique a configuração da GPU:

python -c "import tensorflow as tf; print(tf.config.list_physical_devices('GPU'))"

Se uma lista de dispositivos de GPU for retornada, significa que você instalou o TensorFlow com sucesso.

Windows WSL2

1. Requisitos do sistema

• Windows 10 versão 19044 ou superior (64 bits). Isso corresponde à versão 21H2 do Windows 10, a atualização de novembro de 2021.

Consulte os seguintes documentos para:

• Baixe a atualização mais recente do Windows 10 .
• Instalar o WSL2
• Configurar suporte para GPU NVIDIA® no WSL2

2. Configuração da GPU

Você pode pular esta seção se executar o TensorFlow apenas na CPU.

Instale o driver da GPU NVIDIA, caso ainda não o tenha feito. Você pode usar o seguinte comando para verificar se ele está instalado.

nvidia-smi

3. Instale o TensorFlow

O TensorFlow requer uma versão recente do pip, portanto, atualize sua instalação do pip para garantir que você esteja executando a versão mais recente.

pip install --upgrade pip

Em seguida, instale o TensorFlow com o pip.

# For GPU users
pip install tensorflow[and-cuda]
# For CPU users
pip install tensorflow

4. Verifique a instalação

Verifique a configuração da CPU:

python3 -c "import tensorflow as tf; print(tf.reduce_sum(tf.random.normal([1000, 1000])))"

Se um tensor for retornado, significa que você instalou o TensorFlow com sucesso.

Verifique a configuração da GPU:

python3 -c "import tensorflow as tf; print(tf.config.list_physical_devices('GPU'))"

Se uma lista de dispositivos de GPU for retornada, significa que você instalou o TensorFlow com sucesso.