1. 程式人生 > >Ubuntu 16.04+Gtx1050Ti+cuda 8.0+cudnn 5.1 tensorflow 安裝

Ubuntu 16.04+Gtx1050Ti+cuda 8.0+cudnn 5.1 tensorflow 安裝

現在tensorflow 終於支援高版本(cuda 8.0)一鍵安裝了,今天嘗試了一下,很爽。不用折騰那麼多東西。

建議安裝 anaconda ,一個特別好的Python包管理器,清華大學的源有其映象,安裝速度,更新也挺快的。 其獨有的虛擬環境可以將一個個環境隔離開來,互不影響,這個特別好。

我們首先安裝anaconda ,這個官網有教程,整個安裝過過程傻瓜化。 首先建立虛擬環境: conda create -n tensoflow python=3.7 conda create -n tensoflow python=3.4
conda create -n tensoflow python=3.5

根據習慣選一個Python版本。

tf的gpu版本現在支援直接用pip直接安裝,前提是安裝好cuda8.0和cudnnv5.1(必須是8.0和5.1不然會報錯),然後直接在命令列下pip install tensorflow-gpu

這裡說一下cuda 的安裝,官網給出的安裝教程前提是沒有安裝官方N卡驅動的。按照官網方法一步步來可以安裝成功,並且很快,不需要選擇就可以安裝好,然後根據CUDA 的sample進行測試。一般都會安裝成功

如果之前單獨安裝過N卡的驅動,就不要按照官網的來進行安裝了,安裝的時候,後面不寫引數,然後到第二步驟的時候,選擇不安裝驅動,如果選擇安裝的話,會出錯。