環境
- GTX1080Ti
- Ubuntu14.04LTS(阿里的源)
步驟
- 安裝CUDA8.0
- 安裝Cudnn
- 安裝pip
- 安裝Tensorflow
安裝CUDA8.0
總體上的思路參考這篇文章(ubuntu14.04下tensorflow安裝指南(GPU)),但是其中坑很多。
- 下載好安裝文件,注意,這個安裝文件官方有一版本是錯誤的,現在已經修復。文件名都是cuda_8.0.61_375.26_linux.run,window下是1.4G大小那個是對的。否則直接安裝不了。
- 安裝時,最好不要選擇安裝opengl,會和ubuntu沖突不能再次打開圖形界面了!!!
- 安裝完成后,可能想切換成圖形界面,碰到登陸循環的問題,解決方案網上一大堆,但是都沒去試。最后我是root下卸載了nvidia的驅動重啟就直接成功了。具體原因因為一開始裝ubuntu后沒有顯卡驅動,字都很大模糊,然后先裝了驅動。
nvidia-uninstall
坑主要三個,就夠受的了。后來問題解決了,找到一篇文章,不錯,作者也是用心解決的,不像那些隨便抄抄抄的。ubuntu16.04+cuda8.0+cudnn5.1+caffe 安裝(基于獨立主機&集成顯卡>X1080Ti)
安裝Cudnn
這個好像沒什么太多的坑,照著第一個文章走沒有什么問題。唯一不確定的是在/usr/local/目錄下有兩個,一個cuda還有一個cuda-8.0,最后我選擇兩個都復制了一遍。
安裝pip
本身ubuntu就有python2和3,Tensorflow好像基于2的。
sudo apt-get install python-pip
sudo apt-get install python-dev
安裝Tensorflow
安裝Tensorflow其實很簡單,被網上很多的弄得很麻煩,安裝網上的一些版本他們都去尋找cuda7.0或者7.5的,所以還是讓他自己下載安裝吧。對了,網上的那種還要翻墻。
sudo apt-get install tensorflow
測試
>>> import tensorflow as tf
>>> hello = tf.constant('Hello, TensorFlow!')
>>> sess = tf.Session()
>>> print sess.run(hello)
Hello, TensorFlow!
>>> a = tf.constant(10)
>>> b = tf.constant(32)
>>> print sess.run(a+b)
42