今天因为要用keras跑代码,发现keras是在CPU而非是在GPU上跑,于是下载了tensorflow-gpu和CUDA以及cuDNN,但import tensorflow时还是出现了
no module named ‘tensorflow’
去pycharm上看了下anaconda的环境,结果发现目前tensorflow官方显示发布的最新版本是2.5,但是pip给我安装的是2.6,问题就出在了这里。
解决方案
卸载tensorflow-gpu 2.6,下载tensorflow-gpu 2.5,再次运行代码,完美解决。