1. 準備工作
確認自己的電腦上安裝了nividia驅動,以及cuda,cudnn,具體安裝方法在另一片文章中有介紹
2. pip安裝tensorflow
ubuntu16自帶的Python2.7 和Python3.4,最好使用python3,運行
pip3 install tensorflow-gpu
可能的坑
ImportError:libcudnn.so.6.0:cannot open shared object file:No such file or directory
打開/usr/local/cuda/lib64,發(fā)現(xiàn)現(xiàn)存的文件為libcudnn.so.5.0
tensorflow-gpu最新版本是1.3,需要cuda8以及cudnnv6的支持,而筆者的電腦上安裝的為cuda8和cudnnv5,如果不想更新cudnn,就需要安裝gensorflow-gpu1.2版本
pip3 uninstall tensorflow-gpu
pip3 install tensorflow-gpu==1.2
再次打開python3,輸入import tensorflow,就不會報錯了
3. 最后提醒
不要使用源碼安裝!不要使用源碼安裝!不要使用源碼安裝!
筆者使用源碼安裝了三遍,折騰了一整晚,中間各種坑,最后也沒安裝成功。最好使用pip安裝,一條命令搞定。是有多跟自己過不去非得用源碼安裝。
4. 更新
首先,感謝天瑞老哥,用了整整一中午幫我解決問題。
這一次,嘗試在anaconda_python2.7環(huán)境中配置tensorflow.
1)安裝anaconda
在conda官網(wǎng)下載對應python2.7版本的安裝包,或前往鏡像網(wǎng)站下載對應版本
[]!https://mirrors.tuna.tsinghua.edu.cn/anaconda/archive/
python2.7對應Anaconda2-4.3.1
bash 安裝包路徑
添加路徑
echo 'export PATH="~/anaconda2/bin:$PATH"' >> ~/.bashrc
source ~/.bashrc
2)conda環(huán)境管理
創(chuàng)建python2.7的運行環(huán)境,這里默認安裝python2.7,如安裝python3.4則需加上python==3.4
conda create --name tensorflow_py27
安裝成功后,激活conda環(huán)境
source activate tensorflow_py27
3)安裝tensorflow
pip install tensorflow_gpu
安裝完成進入python后,發(fā)現(xiàn)問題來了:
File "/home/*/.local/lib/python2.7/site-packages/tensorflow/python/client/client_lib.py", line 35...
ImportError: libcudart.so.7.0: cannot open shared object file: No such file or directory
之后,無論筆者安裝哪個tensorflow版本,都會報這個錯誤。
原因出在.local上,發(fā)現(xiàn)此時并沒有在conda文件夾下運行,這說明之前在本地可能安裝過另一個版本的tensorflow。在home目錄下執(zhí)行
pip uninstall tensorflow
原來n年前安裝過tensorflow0.5。。。
好啦,之后再在conda下安裝tensorflow就沒問題了