Project import generated by Copybara

[copybara-service[bot]]

trafficstars

Project import generated by Copybara

[neel04]

@nashid You haven't installed CUDA, which is required to run TensorFlow ops in GPU and utilize its computational cores. However, CUDA is proprietary to Nvidia (whereas AMD uses OpenCL). So in a nutshell, you need CUDA for running TensorFlow, and to run CUDA, you have to have an Nvidia GPU. If you are not willing to buy a GPU, you can always use "Colab" by Google which provides free GPU resources.

Since TF cannot find your GPU, it automatically switches to CPU which takes a lot of time to train on. Hence the high CPU usage and low GPU usage