Home > database >  Keras on GPU on windows with python 3.7.13
Keras on GPU on windows with python 3.7.13

Time:07-04

I had to reinstall completely my python distribution lately and for some reasons I cannot run keras on GPU anymore.

I followed the instructions from Can I run Keras model on gpu? but for some reason, I do not see my GPU when trying to list the devices.

my versions are :

  • tensorflow & tensorflow-gpu : 2.3.0
  • keras : 2.3.1
  • cudatoolkit : 11.3.1

I have not installed cudnn yet as the instructions are a bit blurry for me : do I have to install it in the cudatoolkit directory ? Is it required to run on GPU ?

Thanks

CodePudding user response:

OK so here is the solution :

Python 3.7.13 requires tensorflow-gpu 2.1.0 to detect gpu. So the correct set of versions is :

  • Python 3.7.13
  • tensorflow 2.1.0
  • keras 2.3.0
  • cudnn 7.6.5
  • cudatoolkit 10.1.243

with this it works. Be careful that for some reason, in my case tensorflow downloaded tensorflow-estimator 2.6.0. I had to downgrade to 2.1.0 to get spyder running.

  • Related