TensorFlow 安裝 - CPU

既然要學(xué)習(xí) TensorFlow，那么就要首先學(xué)會如何安裝它。TensorFlow 可以大致地被分為 CPU 版本和GPU 版本，顧名思義，CPU 版本的 TensorFlow 就是在 CPU 之上運(yùn)行的，而 GPU 版本的 TensorFlow 就是使用 GPU 來進(jìn)行相關(guān)的工作的。

那么如何選擇使用的版本呢？我們先來看一下它們之間的差異：

• CPU 版本的 TensorFlow 安裝簡單，幾乎不需要配置，調(diào)用迅速，但是訓(xùn)練模型的速度較慢，適合新手學(xué)習(xí)較為簡單的實(shí)例。
• GPU 版本的 TensorFlow 安裝比較復(fù)雜，配置也比 CPU 版本復(fù)雜很多，而且調(diào)用速度較慢，但是它可以使用GPU的矩陣運(yùn)算，因此它的訓(xùn)練速度比CPU版本的快很多，適用于專業(yè)者。

那么現(xiàn)在這節(jié)課我們就首先來學(xué)習(xí)如何安裝 CPU 版本的 TensorFlow，我們會分別介紹 Windows 操作系統(tǒng)與 Linux 操作系統(tǒng)和 Mac OS 操作系統(tǒng)下如何安裝 TensorFlow-CPU 版本。

1. Window 下 TensorFlow-CPU 的安裝

1.1 準(zhǔn)備工作

首先我們需要在安裝之前進(jìn)行一些簡單的準(zhǔn)備工作，具體如下：

• 官方推薦使用 Windows 7 (64-bit) 及其以后的系統(tǒng)；
• 系統(tǒng)上需要安裝 Python，我們推薦安裝 Python 3.5 到 Python 3.7 的版本（官方的文檔要求），這樣可以避免很多不兼容的問題的發(fā)生；
• 需要熟悉 pip 工具的基本的使用方法，因?yàn)檫@是我們安裝的一個主要工具。
• 確定自己需要使用的 TensorFlow-CPU 版本。

Tips：推薦 TensorFlow2.0 - TensorFlow2.2 版本的任何一個即可，因?yàn)?CPU 版本產(chǎn)生兼容性問題的可能性較小。

1.2 安裝步驟

其實(shí) CPU 版本的安裝非常簡單，只需要兩個步驟即可：

運(yùn)行命令

pip install tensorflow

或者：

pip install tensorflow-cpu

第二個命令是強(qiáng)制安裝 CPU 版本，用來避免 CPU 版本與 GPU 版本的捆綁安裝。

上面是直接使用 pip 安裝，并且使用官方給的版本，如果要指定安裝版本（比如2.0版本），我們可以使用命令：

pip install tensorflow==2.0

因?yàn)?pip 默認(rèn)使用國外的源，連接速度比較慢，因此大家可以換成國內(nèi)的源以此來提升下載速度。
可以使用下面的方式將 pip 的源更換為清華大學(xué)的源來安裝 TensorFlow：

pip install -i https://pypi.tuna.tsinghua.edu.cn/simple tensorflow==2.0

測試是否安裝成功

在使用 pip 安裝完成后我們需要測試是否安裝成功。我們可以使用 Python 運(yùn)行如下代碼：

import tensorflow as tf
print("Version: " + tf.__version__)

若是輸出結(jié)果為：

Version: 2.2.0

若是輸出以上的結(jié)果，則說明安裝成功， 若是沒有輸出版本并且報(bào)錯，則需要查看自己的安裝命令是否正確或者嘗試降低一下安裝的版本。

2. Linux 與 Mac OS 系統(tǒng)下 TensorFlow-CPU 的安裝

因?yàn)槲覀兌紩褂?pip 進(jìn)行安裝，因此 Windows 與 Linux 以及 MaxOS 的 TensorFlow-CPU 版本的安裝大同小異。

首先確保機(jī)器上的 Python 版本為 3.5 版本或以上，如果不是則需要進(jìn)行相應(yīng)的升級。

升級更新自己的 pip，官方要求的 pip 版本為 19.0 及其以上，因此如果自己的 pip 版本沒有達(dá)到要求那么就需要更新一下 pip。

Linux 可以使用以下命令來進(jìn)行升級：

sudo apt-get install python-pip

MacOS 可以使用以下命令來進(jìn)行升級：

sudo easy_install --upgrade pip

在這里只是提供一個更新方法，大家可以使用各種方法更新都可以。

運(yùn)行命令安裝 TensorFlow：

pip install tensorflow-cpu

測試是否安裝成功：

import tensorflow as tf
print("Version: " + tf.__version__)

若是輸出安裝的版本號則標(biāo)識安裝成功，若是報(bào)錯則說明自己安裝失敗，則需要查看自己的當(dāng)前的環(huán)境是否正確。

3. TensorFlow 安裝視頻演示

我在安裝的時(shí)候遇到了幾個小問題在這里給大家說明一下：

1. 沒有 C/C++ 的運(yùn)行環(huán)境，如果遇到這個問題大家可以去這個網(wǎng)址下載一下 Microsoft Visual C++ 然后安裝后重啟一下就可以了。

2. 我安裝的是 TensorFlow2.0 版本，安裝成功之后導(dǎo)入報(bào)錯：

ImportError: DLL load failed: 動態(tài)鏈接庫(DLL)初始化例程失敗

這是因?yàn)槲沂褂玫碾娔X處理器比較老的原因。我把之前的 TensorFlow2.0 版本卸載掉之后重新安裝了 1.5 的版本，并導(dǎo)入成功：

卸載TensorFlow ：pip uninstall tensorflow
安裝1.5 版本：pip install tensorflow==1.5

3. 小結(jié)

總的來說，使用 pip 來安裝 CPU 版本的 TensorFlow 并不算難，只要我們保證 Python 與 pip 版本正確就基本不會出現(xiàn)一些很大的問題。

正確安裝是日后正常使用的保障，因此如果安裝的話一定要仔細(xì)喲。