TensorFlow的安裝(MAC系統(tǒng)CPU版本 & Windows系統(tǒng)GPU版本)
首先,TensorFlow的安裝要求Python必須是64位的扁掸,如圖
TensorFlow的安裝分為CPU版本與GPU版本:
CPU版本:程序可以在CPU上計算或運行碟绑,默認在CPU上贱案,所有的計算機都有CPU
GPU版本:程序既可以在CPU上也可以在GPU上計算或運行,默認是在GPU上運行厚棵,可以指定程序在CPU上運行尤仍,GPU上的運算速度比CPU快箫津,所以一般條件滿足的話盡量安裝GPU的版本
安裝GPU版本的前提條件:需要機器的顯卡必須是NVIDIA的顯卡
對于Python3.5或Python3.6,可以使用pip install tensorflow或pip install tensorflow-gpu直接安裝吓著,對于python2.7版本的Windows系統(tǒng)鲤嫡,只能通過源碼編譯來安裝送挑,其他系統(tǒng)比如Mac或Linux也可以直接通過命令來安裝
CPU版本的安裝可以通過指定TensorFlow版本的形式安裝绑莺,例如:
Pip install tensorflow==1.8.0
安裝成功之后通過pip list可以看到tensorflow與tensorflow-tensorboard就代表已經(jīng)安裝上了,但是還不一定成功惕耕,我們需要通過在python環(huán)境下import tensorflow來測試纺裁,如果沒有異常就證明安裝成功了,如果有異常還需要進一步的處理
CPU版本的安裝相對比較簡單司澎,如果是第一次安裝GPU版本的話欺缘,此時就會出現(xiàn)異常
根據(jù)提示,需要我們下載CUDA(這是NVIDIA提供給程序使用GPU的接口)挤安,畢竟是使用人家的東西谚殊,所以需要專門的接口。
CUDA下載安裝鏈接:https://developer.nvidia.com/cuda-toolkit-archive
CUDA這里有很多的版本蛤铜,具體的選擇根據(jù)異常的提示來嫩絮,比如上文的例子就是需要下載8.0版本的CUDA
之后丛肢,我們還需要下載一個cuDNN,鏈接為:https://developer.nvidia.com/rdp/cudnn-archive
cuDNN版本的選擇與CUDA版本對應(yīng)即可
CUDA的安裝步驟大概如下(默認路徑剿干,最后選擇精簡):
cuDNN的文件下載后直接解壓蜂怎,將里面的三個文件夾直接覆蓋到CUDA的安裝路徑下即可
最后,我們再次import tensorflow就不會出現(xiàn)異常了置尔,我們就成功啦杠步!
如果文字版的教程大家還是不太明白,我也錄制了對應(yīng)的視頻版榜轿,里面講得更加詳細幽歼,干貨更多,每一步的操作大家都可以看得明明白白谬盐,歡迎下載试躏!
04、TensorFlow的安裝CPU版本_MAC系統(tǒng):下載鏈接:https://pan.baidu.com/s/11-tjRr8SQPGGfURGLAW7oQ 密碼:ko7r
05颠蕴、TensorFlow的安裝GPU版本_Windows系統(tǒng):下載鏈接:https://pan.baidu.com/s/12xTGJUhEgSMPyjflkB_Jxw 密碼:rjjr