Python2.7 安装tensorflow-gpu后,import tensorflow出错?
windows的cmd命令行:
pip install --upgrade tensorflow-gpu
安装成功后,不论是Python IDLE还是pycharm均无法导入import tensorflow,提示「ImportError: No module named tensorflow」,pycharm里也没有tensorflow这个模块,请问什么解决?
猜想你的电脑有多个python 版本造成的。
可能是如下原因:
gpu版的tensorflow,除通过命令行安装外,还需要安装CUDA和与之对应的cudnn,注意这两者的版本还需和tensorflow的版本对应,举个例子,我的tensorflow是1.5.0,CUDA是9.0,cudnn是7.0。首先可以在Pycharm里面的setting 中看一下自己python版本以及已经安装后的包,
tensorflow-gpu 需要安装对应的CUDA和CUDNN,对于一个重装系统好多次的人来说,cuda和cudnn版本的很重要!!!两个彼此要对应,而且也要符合python2.7的版本。
PS:刚开始转手ubuntu,其实感觉挺好用的
命令行输入pip list
看看是否安装成功
还有cuda是否安装成功
tensorflow1.6版本支持的cuda是9.0,cudnn是7。所以旧版本的直接升级可能导致出错(1.5开始支持cuda9.0)
(知乎首答献给楼主~)
可能但不限于是:
path有问题
相应的N卡配置文件没有放到正确的路径
推荐上官网看教程,step by step
http://www.tensorflow.org
用virtualenv分别建立不同的虚拟环境,爽的一比
推荐阅读: