如何在CentOS 8上安裝TensorFlow

TensorFlow是Google構建的開放源代碼的機器學習平台。 它可以在不同設備上的CPU或GPU上運行,並且被許多組織(包括Twitter,PayPal,Intel,Lenovo和Airbus)使用。

TensorFlow可以在系統範圍內,Python虛擬環境中,作為Docker容器或與Anaconda一起安裝。

本教程說明了如何在CentOS 8上安裝TensorFlow。

TensorFlow支持Python 2和3。我們將使用Python 3並在虛擬環境中安裝TensorFlow。 虛擬環境允許您在一台計算機上擁有多個不同的隔離Python環境,並在每個項目的基礎上安裝模塊的特定版本,而不必擔心會影響您的其他Project。

在CentOS上安裝TensorFlow#

與其他Linux發行版不同,CentOS 8上默認未安裝Python。要在CentOS 8上安裝Python 3,請以root或sudo用戶身份在終端中運行以下命令:

sudo dnf install python3

上面的命令將安裝Python 3.6和pip。 要運行Python 3,您需要輸入 python3 明確地,並運行點子類型 pip3

從Python 3.6開始,建議的創建虛擬環境的方法是使用 venv 模塊。

導航至您想要存儲TensorFlow項目的目錄。 它可以是您的主目錄,也可以是用戶具有讀寫權限的任何其他目錄。

為TensorFlow項目創建一個新目錄並切換到該目錄:

mkdir tensorflow_project cd tensorflow_project

在目錄中,運行以下命令以創建虛擬環境:

python3 -m venv venv

上面的命令創建一個名為 venv,其中包含Python二進制文件的副本,pip標準Python庫以及其他支持文件。 您可以為虛擬環境使用任何名稱。

要開始使用虛擬環境,請通過鍵入以下內容將其激活:

source venv/bin/activate

激活後,虛擬環境的bin目錄將被添加到 $PATH 可變的。 另外,您的Shell提示符將更改,並顯示您當前正在使用的虛擬環境的名稱。 在這種情況下, venv

TensorFlow安裝要求 pip 版本19或更高版本。 運行以下命令進行升級 pip 到最新版本:

pip install --upgrade pip

現在已經創建並激活了虛擬環境,使用以下命令安裝TensorFlow庫:

pip install --upgrade tensorflow

如果您擁有專用的NVIDIA GPU,並希望利用其處理能力,而不是 tensorflow,安裝 tensorflow-gpu 軟件包,其中包括GPU支持。

在虛擬環境中,可以使用以下命令 pip 代替 pip3python 代替 python3

要驗證安裝,請運行以下命令,該命令將打印TensorFlow版本:

python -c 'import tensorflow as tf; print(tf.__version__)'

在撰寫本文時,TensorFlow的最新穩定版本是2.1.0:

2.1.0

您的TensorFlow版本可能與此處顯示的版本不同。

如果您不熟悉TensorFlow,請訪問TensorFlow入門頁面並了解如何構建第一個ML應用程序。 您還可以從Github克隆TensorFlow模型或TensorFlow-Examples存儲庫,並探索和測試TensorFlow示例。

完成工作後,請輸入以下內容停用環境 deactivate,您將返回到常規shell。

deactivate

而已! 您已經成功安裝了TensorFlow,就可以開始使用它了。

結論#

我們已經向您展示了如何在CentOS 8的虛擬環境中安裝TensorFlow。

如果您遇到問題或有反饋,請在下面發表評論。

python centos

Sidebar