tf-nightly-gpu 和 Keras答案

【问题标题】：tf-nightly-gpu and Kerastf-nightly-gpu 和 Keras
【发布时间】：2020-12-08 02:41:08
【问题描述】：

所以，我很幸运地得到了 RTX 3070。不幸的是，当谈到 tensorflow 时，这并不像我希望的那样工作。我在 google 上花了一些时间，据我所知，tf-nightly-gpu 是解决我的问题的方法。

我已经安装了 Cuda 11/10、cuDNN 和 tf-nightly-gpu，但我无法完成这项工作。我是这里的菜鸟，所以我希望我只是在做一些愚蠢的事情。安装 tensorflow 后我没有收到此错误，但我的代码也没有在当前版本中使用 GPU（术语？）。我的理解是 tf-nightly-gpu 是我的 tensorflow 安装。以下是我的内容：

from keras.models import Sequential
from keras.layers import Dense

这是错误：

C:\Users\Grimace\PycharmProjects\Evo\venv\Scripts\python.exe C:/Users/Grimace/PycharmProjects/Evo/HouseTest.py
Traceback (most recent call last):
  File "C:\Users\Grimace\PycharmProjects\Evo\venv\lib\site-packages\keras\__init__.py", line 3, in <module>
    from tensorflow.keras.layers.experimental.preprocessing import RandomRotation
ModuleNotFoundError: No module named 'tensorflow.keras.layers'

During handling of the above exception, another exception occurred:

Traceback (most recent call last):
  File "C:/Users/Grimace/PycharmProjects/Evo/HouseTest.py", line 2, in <module>
    from keras.models import Sequential
  File "C:\Users\Grimace\PycharmProjects\Evo\venv\lib\site-packages\keras\__init__.py", line 6, in <module>
    'Keras requires TensorFlow 2.2 or higher. '
ImportError: Keras requires TensorFlow 2.2 or higher. Install TensorFlow via `pip install tensorflow`

【问题讨论】：

您是否尝试过 pip install tensorflow 之类的错误提示？
你是否安装了 tensorflow 作为后端？ pip isntall tensorflow
正如其他人所说，tensorflow 现在在库中内置了 GPU 支持，无需安装 tensorflow-gpu 或 tf-nightly-gpu。我遵循以下关于数据科学的指南，但最后我安装了 tensorflow 而不是 tensorflow-gpu。这对我有用towardsdatascience.com/…
RTX 3070 需要 CUDA 11.x 和使用 10.x 的 tensorflow 存在问题。这就是我安装 tf-nightly-gpu 的原因。 tf-nightly-gpu 是完整的张量流，我不正确吗？我在想 Keras 正在寻找 tensorflow 而不是使用 tf-nightly ...某种类型的硬编码依赖问题。当我安装当前版本的 tensorflow 及其依赖项时，它不适用于我的 GPU。
安装tf-nightly-gpu后可以导入tensorflow吗？它可以正常工作吗？

标签： python tensorflow keras

【解决方案1】：

您正在导入keras。对于 tf-nightly，以及所有最新版本的 tensorflow，请导入 tf.keras。 Tensorflow nightly 和 tf.keras 在我的 3090 和 CUDA 11.1 上运行良好。

【讨论】：