Блог

Главная - Вопросы по программированию - tf-nightly-gpu и Keras

tf-nightly-gpu и Keras

#python #tensorflow #keras

#python #тензорный поток #keras

Вопрос:

Итак, мне повезло, и я получил в свои руки RTX 3070. К сожалению, это работает не так хорошо, как мне бы хотелось, когда дело доходит до tensorflow. Я провел некоторое время в Google, и, насколько я могу судить, tf-nightly-gpu — это решение моих проблем здесь.

Я установил Cuda 11/10, cuDNN и tf-nightly-gpu, но мне не удается заставить это работать. Я здесь новичок, поэтому я надеюсь, что я просто делаю что-то глупое. Я не получаю эту ошибку при установленном tensorflow, но мой код также не использует графический процессор в текущей сборке (терминология?). Я понимаю, что tf-nightly-gpu — это моя установка tensorflow. Вот мои включает:

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

Вот ошибка:

 C:UsersGrimacePycharmProjectsEvovenvScriptspython.exe C:/Users/Grimace/PycharmProjects/Evo/HouseTest.py
Traceback (most recent call last):
  File "C:UsersGrimacePycharmProjectsEvovenvlibsite-packageskeras__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:UsersGrimacePycharmProjectsEvovenvlibsite-packageskeras__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`

Комментарии:

1. вы пробовали то, что упоминается в ошибке, например, pip install tensorflow?

2. Вы установили tensorflow в качестве серверной части?? pip — это не весь тензорный поток

3. Как заявляли другие, tensorflow теперь имеет встроенную в libray поддержку GPU без необходимости установки tensorflow-gpu или tf-nightly-gpu. Я следовал следующему руководству по направлению к науке о данных, но в самом конце я установил tensorflow в отличие от tensorflow-gpu. Это заставило меня работать towardsdatascience.com /…

4. Существуют проблемы с RTX 3070, требующим CUDA 11.x и tensorflow с использованием 10.x. Это то, что заставляет меня устанавливать tf-nightly-gpu. Разве я не прав, что tf-nightly-gpu является полным тензорным потоком? Я думал, что Keras ищет tensorflow и не использует tf-nightly … какая-то проблема с жестко запрограммированной зависимостью. Когда я устанавливаю текущую сборку tensorflow и ее зависимостей, она не работает с моим графическим процессором.

5. Можете ли вы импортировать tensorflow после установки tf-nightly-gpu? Работает ли это без проблем?

Ответ №1:

Вы импортируете keras . Пожалуйста, импортируйте tf-nightly и все последние версии tensorflow tf.keras . Tensorflow работает каждую ночь и tf.keras хорошо работает на моем 3090 с CUDA 11.1.