Ubuntu16 安裝gpu模式的tensorflow

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就沒問題了

最后編輯于
?著作權歸作者所有,轉載或內容合作請聯(lián)系作者
【社區(qū)內容提示】社區(qū)部分內容疑似由AI輔助生成,瀏覽時請結合常識與多方信息審慎甄別。
平臺聲明:文章內容(如有圖片或視頻亦包括在內)由作者上傳并發(fā)布,文章內容僅代表作者本人觀點,簡書系信息發(fā)布平臺,僅提供信息存儲服務。

友情鏈接更多精彩內容