一開始花了幾天在linux上重新編譯tensorflow,遇到各種各樣的問題,而且耗費(fèi)時間非常長,還是在最后一步失敗了.
這個故事告訴我們纱新,最好先別直接編譯tensorflow枫笛。
由于我現(xiàn)在用的是CUDA9.2,而目前tf 的official pip package最新的還是cuda 9.0,所以后來我決定把cuda版本下降至9.0,然后再配置,最后終于成功了.
1,卸載cuda9.2
sudo apt-get remove cuda
sudo apt-get autoclean
sudo apt autoremove cuda(選)
sudo apt-get remove nvidia*(選)
2,接下來按照下面的網(wǎng)址配置就可以完成
http://www.python36.com/install-tensorflow-using-official-pip-pacakage/
(注意最好配置編譯好的版本朋沮,否則會很痛苦)
3,問題
0浅妆,CUDA文件烟勋,和cudnn下載極端緩慢
我們竟然可以直接復(fù)制鏈接启摄,到迅雷下載上下,又塊又穩(wěn)定实束。
1奥秆,在step6 中安裝步驟會和cuda官網(wǎng)上的安裝步驟沖突,這里結(jié)合使用下面命令(deb local):
sudo dpkg -i cuda-repo-ubuntu1604-9-0-local_9.0.176-1_amd64.deb
sudo apt-key add /var/cuda-repo-/7fa2af80.pub
sudo apt-get update
sudo apt-get install cuda-9-0(這里不能使用sudo apt-get install cuda,否則會安裝最新版本)
但是這里還會出現(xiàn)mismatching的問題,導(dǎo)致無法安裝.最后發(fā)現(xiàn)的問題是我之前配置了很多國內(nèi)的鏡像源,我把這些源刪除之后重新安裝就可以通過了.
sudo cp /etc/apt/sources.list /etc/apt/sources.list.bak
sudo gedit /etc/apt/sources.list #修改源
sudo apt-get update
sudo apt-get upgrade
在step 9 中,并沒用cuDNN 7.1.2的相關(guān)下載,這里降級為cuDNN 7.0.5 版本.