我的 keras 后端 tensorflow 不使用 gpu？

Posted 2023-02-16

【中文标题】我的 keras 后端 tensorflow 不使用 gpu？

【英文标题】：My keras backend tensorflow does not use the gpu?

【发布时间】：2020-06-13 18:31:37

【问题描述】：

我先安装 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

【问题讨论】：

尝试使用pycharm，这样您就可以获得指定版本的TF和keras，并尝试运行2.0而不是2.1，它将解决您的问题

【参考方案1】：

尝试使用 Tensorflow-gpu=2.0.0 和 Keras=2.3.1 这将解决您的问题。