What worked for me was
pip uninstall tensorflow
Then I installed it using this
Check your gpu setup and install accordingly. Should work fine.