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


推薦閱讀:
相關文章