My keras backend tensorflow does not use the gpu?

Viewed 304

I install the tensorflow gpu first

pip install tensorflow-gpu. 
pip install keras

but when I;m running the gpu task. it does not run with the gpu.
It run with CPU.

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

try to user Tensorflow-gpu=2.0.0 and Keras=2.3.1 this will solve your problem.

Related