Join GitHub today
GitHub is home to over 28 million developers working together to host and review code, manage projects, and build software together.Sign up
Installing keras makes tensorflow can't find GPU #5776
I have used keras + tensorflow-gpu in my old computer, it's very ok. (I forget the tensorflow version)
I install keras with tensorflow-gpu (version 1.0.1) in my new computer, and before install keras, tensorflow can find my GPU. But after install keras, tensorflow only can find CPU.
I use these code to check GPU:
And I tried to install keras from source ($ python setup.py install), it would have some error and install fail.
Is this problem about version ?
Yeah, currently when you install Keras using pip or pip3 it blows off existing TF and installs the default, non-GPU version. It'd be great if there was a flag to not touch existing TF.
The workaround is to uninstall TF after installing Keras, and then installing the GPU version using pip or pip3 depending on your preferred python version. Not very elegant, but you gotta do what you gotta do.