Как импортировать tensorflow-gpu

Появилась возможность использоваться tensorflow-gpu. Сделал все как положенно (https://stackoverflow.com/questions/51306862/how-to-use-tensorflow-gpu). Но при импорте библиотеки выдает

ModuleNotFoundError: No module named 'tensorflow'

Как к GPU версии обращаться? Мне для начала надо сделать банальные вещи, и могу работать на cpu, но все же, если есть возможность, надо ее реализоваться