Running on GPU

[OOps717]

Hello, I am trying to run training on both my computer and google colab, however even if I am trying to change something it still trains on CPU. I am guessing if there is any solution ?

[kheniparth]

I did face that issue and found that you would have to remove tensorflow package and install tensorflow-gpu package for python and then the code will start utilizing GPU RAM. But my CPU is still getting utilized 100% not sure, why but I can confirm that the training rate is increased exponentially.

Note: tensorflow-gpu package will require some extra configuration to be done or else you will start getting errors, also you would have to you python 3.6 env or else this code breaks (learnt this the hard way )

[Pawandeep-prog]

The solution i am going to share worked for me

first,

unistall tensorflow from environment
**pip uninistall tensorflow**

then,

install tensorflow-gpu 1.14
**pip install tensorflow-gpu==1.14**