我正在尝试使用 docker 镜像将 Keras 与 tensorflow-gpu 结合使用https://github.com/floydhub/dl-docker但是 keras 使用的是 cpu 版本的 tensorflow。我有 cpu 和 gpu 版本的 tensorflow
中央处理器
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
图形处理器
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
所以我在这个线程中读到它告诉我卸载 tensorflow 所以它将使用 tensorflow-gpu。但是在我卸载 tensorflow 并运行代码后它告诉我没有名为 tensorflow 的模块
答案1
我在安装时也遇到了这个问题。我所做的就是卸载 Tensorflow GPU。然后您可以尝试重新安装它。确保您的图形驱动程序也使用 CUDNN 和 Cuda 9.0。