#python-3.x #tensorflow
#python-3.x #tensorflow
Вопрос:
Версия Tensorflow — 1.0.1 Версия GPU — 1.0.1 python — 3.6 Graphics (Ubuntu 18.04) — Intel® Sandybridge Desktop Мы пытаемся запустить код python с использованием Tensorflow. Но это было очень медленно. Итак, мы хотели бы запустить код на GPU. Кто-нибудь может подсказать, как использовать GPU для запуска кода python?
Комментарии:
1. Вы могли бы проверить это: используя графические процессоры . Вы также можете попробовать Google Colab, если хотите
2. Для этого есть другой пакет. Если у вас есть все другие предварительные требования, установите его с помощью pip install TensorFlow-gpu вам не нужно изменять свой код, TensorFlow должен обрабатывать графический процессор самостоятельно. Но не все работает быстрее на gpu. Только если вы, например, тренируете сеть (задачи, которые могут выполняться параллельно)
Ответ №1:
Вы можете удалить tensor flow и установить версию, совместимую с GPU, с
pip install tensorflow-gpu
Ответ №2:
config = tf.compat.v1.ConfigProto(gpu_options = tf.compat.v1.GPUOptions(per_process_gpu_memory_fraction=0.8))
config.gpu_options.allow_growth = True
session = tf.compat.v1.Session(config=config)
tf.compat.v1.keras.backend.set_session(session)
Поместите этот код перед вашим кодом python, а также убедитесь, что у вас установлены все библиотеки DLL cuDNN.