tensorflow não está usando gpu e nvidia-smi: comando não encontrado

1

a placa Nvidia do meu sistema: insira a descrição da imagem aqui

Eu quero usar a placa gráfica para o meu tensorflow e eu instalei e re-instalei novamente, mas tensorflow não está usando GPU e eu também instalei meus drivers Nvidia mas quando eu executo nvidi-smi então um comando não é encontrado

minha inicialização segura está desativada Eu configurei nouveau = 0

    
por Simer Wadhwa 08.03.2018 / 07:59

1 resposta

0

Para referência futura, considere adicionar o texto do seu terminal ao seu post de perguntas.

Instale a versão GPU do TensorFlow, fazendo

sudo pip install tensorflow-gpu .

No entanto, você deve ter o CUDA instalado para que o TF-GPU use sua GPU. Você pode baixar o CUDA de aqui . Eu recomendo que você use o CUDA 9.0 com o TensorFlow 1.7

    
por Abien Fred Agarap 03.04.2018 / 20:10