如何在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
代替 pip3
和 python
代替 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