keras - nenhum módulo chamado tensorflow

0

Estou tentando usar o Keras com o tensorflow-gpu usando o docker image link Mas o keras está usando a versão cpu do tensorflow. Eu tenho cpu e gpu versão do tensorflow

cpu

root@08b5fff06800:~# pip show tensorflow
Name: tensorflow
Version: 1.3.0
Summary: TensorFlow helps the tensors flow
Home-page: http://tensorflow.org/
Author: Google Inc.
Author-email: [email protected]
License: Apache 2.0
Location: /usr/local/lib/python2.7/dist-packages
Requires: tensorflow-tensorboard, six, protobuf, mock, numpy, backports.weakref, wheel

gpu

root@08b5fff06800:~# pip show tensorflow-gpu
Name: tensorflow-gpu
Version: 0.12.1
Summary: TensorFlow helps the tensors flow
Home-page: http://tensorflow.org/
Author: Google Inc.
Author-email: [email protected]
License: Apache 2.0
Location: /usr/local/lib/python2.7/dist-packages
Requires: mock, numpy, protobuf, wheel, six

então eu li neste tópico que me diz para desinstalar o tensorflow para que ele use o tensorflow-gpu. mas depois eu desinstalo tensorflow e eu corro um código que me diz nenhum módulo chamado tensorflow

    
por Upvotes All Downvoted Posts 07.09.2017 / 08:32

1 resposta

0

Eu tive esse problema instalando também. O que fiz foi desinstalar o GPU Tensorflow. Então você pode tentar reinstalá-lo. Verifique se você também está usando o CUDNN e o Cuda 9.0 para os drivers gráficos.

    
por Jake Malis 31.07.2018 / 13:47