Мой бэкэнд-поток keras не использует графический процессор?

Question

Сначала я устанавливаю GPU Tensflow

pip install tensorflow-gpu. 
pip install keras

но когда я запускаю задачу gpu. это не работает с GPU. Это работает с процессором.

import keras
import tensorflow as tf
print(keras.__version__)
print(tf.__version__)
2.3.1
2.1.0

Answer 1

попробуйте пользователю Tensorflow-gpu = 2.0.0 и Keras = 2.3.1 это решит вашу проблему.