本指南將介紹如何在 Windows 上安裝 TensorFlow。雖然這些說(shuō)明可能也適用于其他 Windows 版本,但我們只在滿足以下要求的計(jì)算機(jī)上驗(yàn)證過(guò)這些說(shuō)明(而且我們只支持在此類計(jì)算機(jī)上按這些說(shuō)明操作):
- 64 位、x86 臺(tái)式機(jī)或筆記本電腦
- Windows 7 或更高版本
確定要安裝哪種 TensorFlow
您必須從以下 TensorFlow 類型中選擇其一來(lái)進(jìn)行安裝:
- 僅支持 CPU 的 TensorFlow。如果您的系統(tǒng)沒(méi)有 NVIDIA? GPU,則必須安裝此版本豁翎。請(qǐng)注意,此版本的 TensorFlow 通常更容易安裝(用時(shí)通常在 5 或 10 分鐘內(nèi))隅忿,所以即使您擁有 NVIDIA GPU心剥,我們也建議先安裝此版本。預(yù)編譯的二進(jìn)制文件將使用 AVX 指令背桐。
- 支持 GPU 的 TensorFlow优烧。TensorFlow 程序在 GPU 上的運(yùn)行速度通常要比在 CPU 上快得多。因此牢撼,如果您的系統(tǒng)配有滿足以下所示先決條件的 NVIDIA? GPU匙隔,并且您需要運(yùn)行性能至關(guān)重要的應(yīng)用,則最終應(yīng)安裝此版本熏版。
運(yùn)行支持 GPU 的 TensorFlow 所需滿足的要求
如果您要使用本指南描述的其中一種方式安裝支持 GPU 的 TensorFlow纷责,就必須在系統(tǒng)上安裝以下 NVIDIA 軟件:
- CUDA? 工具包 9.0。如需了解詳情撼短,請(qǐng)參閱 NVIDIA 的文檔再膳。請(qǐng)務(wù)必按照 NVIDIA 文檔中的說(shuō)明將相關(guān)的 CUDA 路徑名附加到
%PATH%
環(huán)境變量上。 - 與 CUDA 工具包 9.0 相關(guān)聯(lián)的 NVIDIA 驅(qū)動(dòng)程序曲横。
- cuDNN v7.0喂柒。如需了解詳情,請(qǐng)參閱 NVIDIA 文檔禾嫉。請(qǐng)注意灾杰,cuDNN 通常安裝在與其他 CUDA DLL 不同的位置。請(qǐng)務(wù)必將 cuDNN DLL 的安裝目錄添加到
%PATH%
環(huán)境變量上熙参。 - CUDA 計(jì)算能力為 3.0 或更高的 GPU 卡(用于從源代碼編譯)艳吠,以及 CUDA 計(jì)算能力為 3.5 或更高的 GPU 卡(用于安裝我們的二進(jìn)制文件)。如需了解支持的 GPU 卡的列表孽椰,請(qǐng)參閱 NVIDIA 文檔胡控。
如果您的某個(gè)軟件包不同于上述版本空扎,請(qǐng)改為指定的版本嘱朽。特別是现喳,cuDNN 版本必須完全匹配:如果無(wú)法找到 cuDNN64_7.dll
,TensorFlow 就不會(huì)加載锐涯。要使用不同版本的 cuDNN磕诊,您必須從源代碼構(gòu)建。
確定如何安裝 TensorFlow
您必須選擇安裝 TensorFlow 的方式。目前可支持如下幾種方式:
- “原生”pip
- Anaconda
原生 pip 會(huì)直接在您的系統(tǒng)上安裝 TensorFlow秀仲,而不是通過(guò)虛擬環(huán)境融痛。原生 pip 安裝并未隔離在單獨(dú)的容器中進(jìn)行,因此可能會(huì)干擾系統(tǒng)中其他基于 Python 的安裝神僵。但是,如果您熟悉 pip 和您的 Python 環(huán)境覆劈,通常只需一條命令即可進(jìn)行原生 pip 安裝保礼。此外,如果使用原生 pip 安裝责语,用戶可以從系統(tǒng)上的任何目錄運(yùn)行 TensorFlow 程序炮障。
在 Anaconda 中,您可以使用 conda 來(lái)創(chuàng)建一個(gè)虛擬環(huán)境坤候。但是胁赢,在 Anaconda 內(nèi)部,我們建議使用 pip install
命令來(lái)安裝 TensorFlow白筹,而不要使用 conda install
命令智末。
注意:conda 軟件包是由社區(qū)提供支持的,并沒(méi)有任何官方支持徒河。也就是說(shuō)系馆,TensorFlow 團(tuán)隊(duì)既不測(cè)試也不維護(hù) conda 軟件包。若使用該軟件包顽照,您需要自行承擔(dān)相關(guān)風(fēng)險(xiǎn)由蘑。
使用原生 pip 進(jìn)行安裝
如果您的計(jì)算機(jī)上未安裝以下某個(gè) Python 版本,請(qǐng)立即安裝:
在 Windows 上代兵,TensorFlow 支持 Python 3.5.x 和 3.6.x尼酿。請(qǐng)注意,Python 3 附帶有 pip3 軟件包管理器植影,您需使用此程序來(lái)安裝 TensorFlow裳擎。
要安裝 TensorFlow,請(qǐng)啟動(dòng)終端何乎。然后在該終端中發(fā)出正確的 <tt style="box-sizing: inherit;">pip3 install</tt> 命令句惯。要安裝僅支持 CPU 的 TensorFlow 版本,請(qǐng)輸入以下命令:
C:\> pip3 install --upgrade tensorflow
要安裝 GPU 版本的 TensorFlow支救,請(qǐng)輸入以下命令:
C:\> pip3 install --upgrade tensorflow-gpu
安裝成功后
驗(yàn)證您的安裝
啟動(dòng)終端抢野。
如果您是通過(guò) Anaconda 進(jìn)行安裝,請(qǐng)激活您的 Anaconda 環(huán)境各墨。
從 cmd 中調(diào)用 Python指孤,如下所示:
C:\Users\majia>python
在 Python 交互式 shell 中輸入以下幾行簡(jiǎn)短的程序代碼:
>>> import tensorflow as tf
>>> hello = tf.constant('Hello, TensorFlow!')
>>> sess = tf.Session()
>>> print(sess.run(hello))
如果系統(tǒng)輸出以下內(nèi)容,說(shuō)明您可以開(kāi)始編寫 TensorFlow 程序了:
Hello, TensorFlow!
如果在執(zhí)行代碼中出現(xiàn)
解決辦法:
在最頂行寫入
import os
os.environ['TF_CPP_MIN_LOG_LEVEL'] = '2'
如下圖所示
好了,現(xiàn)在整個(gè)TensorFlow在windows安裝已經(jīng)完成可以開(kāi)始編程了恃轩!