我的 keras 后端 tensorflow 不使用 gpu？

Question

我先安装 tensorflow gpu

pip install tensorflow-gpu. 
pip install keras

但是当我运行 gpu 任务时。它不与 gpu 一起运行。
它使用 CPU 运行。

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 这将解决您的问题。