TensorFlow 是一個開源的、基於 Python 的機器學習框架,它由 Google 開發,並在圖形分類、音頻處理、推薦系統和自然語言處理等場景下有着豐富的應用,是目前最熱門的機器學習框架。是為深度神經網絡(DNN)開發的功能強大的開源軟件庫。除了 Python,TensorFlow 也提供了 C/C++、Java、Go、R 等其它編程語言的接口。
TensorFlow 可以在 Ubuntu 和 macOS 上基於 native pip、Anaconda、virtualenv 和 Docker 進行安裝,對於 Windows 操作系統,可以使用 native pip 或 Anaconda。
TensorFlow安裝准備工作
TensorFlow 安裝的前提是系統安裝了Python 2.5或更高版本,本文是以 Python 3.8.11(Anaconda 3 版)為基礎設計的。
為了安裝 TensorFlow,首先確保你已經安裝了Anaconda。
可以從網址(https://www.continuum.io/downloads)中下載並安裝適用於 Windows/macOS 或 Linux 的 Anaconda。
安裝完成后,可以在窗口中使用以下命令進行安裝驗證:
conda --version
安裝了Anaconda,下一步決定是否安裝TensorFlow CPU 版本或 GPU 版本。幾乎所有計算機都支持 TensorFlow CPU 版本,而 GPU 版本則要求計算機有一個 CUDA compute capability 3.0 及以上的 NVDIA GPU 顯卡(對於台式機而言最低配置為 NVDIA GTX 650)。
CPU 與 GPU 的對比:
中央處理器(CPU)由對順序串行處理優化的內核(4~8個)組成。
圖形處理器(GPU)具有大規模並行架構,由數千個更小且更有效的核芯(大致以千計)組成,能夠同時處理多個任務。
TensorFlow CPU版安裝具體做法
1.在命令行中使用以下命令創建 conda 環境(如果使用 Windows,最好在命令行中以管理員身份執行):
conda create -n tensorflow python=3.8.11
選擇“y”回車,開始下載......
在Anaconda3下的envs文件夾中多了tensorflow文件夾
運行 開始菜單 -> Anaconda3 -> Anaconda Navigator
,點擊左側的Environments
,可以看到tensorflow
的環境已經創建好了。
2.激活 conda 環境:
activate tensorflow
3.該命令應提示:
當不使用tensorflow時,關閉tensorflow環境,命令為:
deactivate
4.根據要在 conda 環境中安裝的 TensorFlow 版本,輸入以下命令:
pip install --upgrade --ignore-installed tensorflow
好慢。如果太慢了,可以提前設置使用輸入清華倉庫鏡像,這樣更新會快一些:
conda config --add channels https://mirrors.tuna.tsinghua.edu.cn/anaconda/pkgs/free/
conda config --set show_channel_urls yes
安裝完成,顯示如下:
5.測試是否安裝成功:
在Anaconda窗口的tensorflow的環境中,跑一個test.py,看是否能運行成功。
如下圖,得到了“Hello World”證明一切安裝順利。
test.py的內容如下:
import tensorflow as tf tf.compat.v1.disable_eager_execution() # Create TensorFlow object called tensor hello_constant = tf.constant('Hello World!') with tf.compat.v1.Session() as sess: # Run the tf.constant operation in the session output = sess.run(hello_constant) print(output.decode())# bytestring decode to string.
上述test.py代碼只是檢測了tensorflow是否安裝成功。
如果想手動聲明讓電腦用CPU運算,則......