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分別建立不同的虛擬環境,爽的一比
推薦閱讀: