如何在 Ubuntu 20.04 上安裝 TensorFlow

Tensorflow 是一個用於機器學習和人工智能的開源平台。 它由 Google Brain 團隊開發。 它包含工具、庫和社區資源,供開發人員構建機器學習驅動的應用程序和深度神經網絡。

TensorFlow 安裝可以在系統範圍內進行,作為 docker 容器,在 Python 虛擬環境中,或使用 Anaconda。 創建虛擬環境並安裝 Tensorflow 是完成工作的有效方式。 虛擬環境允許開發人員在一台機器上處理多個 Python 項目環境,他們還可以在隔離的虛擬項目環境中安裝特定的應用程序版本,而不會中斷其他 Python 項目。 這種方法可能有助於解決與版本依賴相關的問題。

在本教程中,我們將學習如何 安裝 TensorFlowUbuntu 20.04 在 Python 虛擬環境中。

先決條件

  • 一個 Ubuntu 20.04 服務器
  • 最低 4GB 內存
  • Python 3.8 或更高版本
  • 一個用戶 sudo 特權

步驟 1:驗證 Python 安裝

安裝 Python 是 TensorFlow 庫的基本要求。 Python 3.8 已經安裝在 Ubuntu 上。 執行以下命令以顯示已安裝的 Python 版本,以驗證 Python 是否安裝在您的 Ubuntu 系統上:

$ python3 -V

以下輸出顯示在終端上,您可以在其中看到已安裝的 python 版本:

第 2 步:安裝 Python Venv 模塊

建議使用 venv 模塊來創建虛擬環境,該模塊已經包含在 Python3-venv 包中。 可以使用以下命令安裝 venv 模塊:

$ sudo apt install python3-venv python3-dev

安裝 python3-venv 包

第 3 步:為 TensorFlow 項目創建目錄

在您的 Ubuntu 系統上安裝 python3-venv 包後,您將為 TensorFlow 項目創建新的 python 虛擬環境。 虛擬環境必須位於 Ubuntu 發行版的主目錄中。

如果您希望在新目錄中創建虛擬環境,請執行以下“mkdir”命令以創建一個名為 tensorflow_project 的新目錄:

$ mkdir tensorflow_project

使用以下命令,您將導航到我們剛剛在上一節中創建的新目錄:

$ cd tensorflow_project

創建 tensorflow 項目目錄

步驟4: 創建 Python 虛擬環境 (venv) 並激活venv

您的系統現在已準備好創建名為 virtual_env (venv) 的虛擬環境。 使用以下命令,您可以在目錄中創建一個新的 python 虛擬環境:

$ python3 -m venv venv

您可以隨意命名虛擬環境。 前面的命令創建名為的虛擬環境 venv 包含所有 python 庫、pip 包管理器和必要的 python 二進製文件的副本。 要使用虛擬環境,您需要激活 venv 作為:

$ source venv/bin/activate

創建並激活python虛擬環境

第五步:升級python pip 包管理器

PIP 是一個 Python 包管理器,用於安裝和維護可以通過 pip 安裝的包。 PIP 版本 19 或更高版本是安裝 TensorFlow 的先決條件。 通過使用以下命令,您可以將 PIP 升級到 pip 版本 19 或更高版本:

(venv) $ pip install --upgrade pip

升級點

第 6 步:使用 Pip 包管理器安裝 TensorFlow

在這一步中,我們將轉向通過 pip 安裝 TensorFlow。 執行以下命令以使用 pip 包管理器安裝 TensorFlow 庫:

(venv) $ pip install --upgrade tensorflow

這將安裝 TensorFlow 並將其升級到最新版本。

安裝和升級 tensorflow

安裝一個 具體版本 在 TensorFlow 中,輸入:

$ pip install tensorflow==2.6.0

第 7 步: 驗證 TensorFlow 安裝

在您的 Ubuntu 系統上成功安裝 TensorFlow 庫後,將使用以下命令來驗證安裝過程:

(venv) $ python -c 'import tensorflow as tensor_flow; print(tensor_flow.__version__)'

終端上顯示安裝的TensorFlow版本,這裡,我們安裝了 2.7.0 Ubuntu 系統上的 tensorflow 版本。

第 8 步: 停用虛擬環境

使用特定的虛擬環境後,您可以通過運行以下命令將其停用,這將使您回到正常的 shell 環境:

(venv) $ deactivate

結論

我們提供了在 Ubuntu 20.04 系統上安裝 Tensorflow 的分步指南。 我們在本文中討論瞭如何創建一個新的虛擬環境並激活它。 我們在虛擬環境中安裝了 TensorFlow。 最後,我們解釋瞭如何通過使用一個命令來停用 python 虛擬環境。