Índice:

Como faço para que o TensorFlow use minha GPU?
Como faço para que o TensorFlow use minha GPU?

Vídeo: Como faço para que o TensorFlow use minha GPU?

Vídeo: Como faço para que o TensorFlow use minha GPU?
Vídeo: Como Instalar O Tensorflow Com Cuda (GPU Para Deep Learning) 2024, Maio
Anonim

Passos:

  1. Desinstalar sua velho tensorflow .
  2. Instalar tensorflow - gpu pip install tensorflow - gpu .
  3. Instale Nvidia Placa de vídeo & Drivers (provavelmente você já tem)
  4. Baixe e instale CUDA.
  5. Baixe e instale cuDNN.
  6. Verifique por programa simples.

Além disso, o TensorFlow precisa de GPU?

GPU TensorFlow Apoio, suporte requer uma variedade de drivers e bibliotecas. Esta configuração apenas requer o NVIDIA® GPU motoristas. Estas instruções de instalação são para a versão mais recente do TensorFlow . Veja as configurações de construção testadas para as versões CUDA e cuDNN para usar com TensorFlow lançamentos.

Da mesma forma, como faço para usar várias GPUs no TensorFlow? Se um TensorFlow operação tem CPU e GPU implementações, TensorFlow colocará automaticamente a operação para ser executada sobre uma GPU dispositivo primeiro. Se você tem mais de um GPU , a GPU com o menor ID será selecionado por padrão. Contudo, TensorFlow não coloca operações em várias GPUs automaticamente.

Tendo isso em mente, o que é GPU no TensorFlow?

UMA GPU (Unidade de processamento gráfico) é um componente da maioria dos computadores modernos, projetado para realizar cálculos necessários para gráficos 3D.

O TensorFlow pode ser executado na GPU AMD?

Este código pode correr nativamente em AMD bem como Nvidia GPU . Sim é possível executar tensorflow sobre GPU da AMD mas seria um grande problema. Como tensorflow usa CUDA, que é proprietário dele posso 't rodar em GPUs AMD então você precisa usar OPENCL para isso e tensorflow não está escrito nisso.

Recomendado: