Jak nainstalovat TensorFlow na Ubuntu 20.04

TensorFlow je bezplatná a open source platforma pro strojové učení vytvořená společností Google. Používá jej řada organizací, včetně Twitter, PayPal, Intel, Lenovo a Airbus.

TensorFlow lze nainstalovat v rámci celého systému, ve virtuálním prostředí Pythonu, jako Dockercontainer nebo s Anacondou.

Tento tutoriál vysvětluje, jak nainstalovat TensorFlow ve virtuálním prostředí Pythonu na Ubuntu 20.04.

Virtuální prostředí vám umožňuje mít na jednom počítači několik různých izolovaných prostředí Pythonu a instalovat konkrétní verzi modulu pro každý projekt, aniž byste se museli obávat, že to ovlivní vaše další projekty.

Instalace TensorFlow na Ubuntu 20.04 #

Ubuntu 20.04 se standardně dodává s Pythonem 3.8. Můžete ověřit, že je Python 3 na vašem systému nainstalován, zadáním:

python3 -V

Výstup by měl vypadat takto:

Python 3.8.2

Doporučený způsob vytvoření virtuálního prostředí je pomocí venv modul, který je součástí python3-venv balíček.

Chcete-li nainstalovat python3-venv balíček spusťte následující příkaz:

sudo apt install python3-venv python3-dev

Jakmile je modul nainstalován, jste připraveni vytvořit virtuální prostředí pro náš projekt TensorFlow.

Přejděte do adresáře, kam chcete uložit vaše virtuální prostředí Python 3. Může to být váš domovský adresář nebo jakýkoli jiný adresář, kde má váš uživatel oprávnění ke čtení a zápisu.

Vytvořte nový adresář pro projekt TensorFlow a cdindo něj:

mkdir my_tensorflowcd my_tensorflow

V adresáři spusťte následující příkaz k vytvoření virtuálního prostředí:

python3 -m venv venv

Druhý venv je název virtuálního prostředí. Pro virtuální prostředí můžete použít libovolný název.

Výše uvedený příkaz vytvoří adresář s názvem venv , který obsahuje kopii binárky Pythonu, správce balíčků Pip, standardní knihovnu Pythonu a další podpůrné soubory.

Chcete-li začít používat virtuální prostředí, aktivujte jej spuštěním activate skript:

source venv/bin/activate

Po aktivaci bude adresář bin virtuálního prostředí přidán na začátek systému $PATH variabilní. Také se změní výzva shellu a zobrazí název virtuálního prostředí, ve kterém se aktuálně nacházíte. V tomto příkladu je to (venv) .

Instalace TensorFlow vyžaduje pip verze 19 nebo vyšší. Pro upgrade pip zadejte následující příkaz na nejnovější verzi:

pip install --upgrade pip

Nyní, když je virtuální prostředí aktivováno, je čas nainstalovat balíček TensorFlow.

pip install --upgrade tensorflow

Pokud máte vyhrazený grafický procesor NVIDIA a chcete využít jeho výpočetní výkon, místo tensorflow , nainstalujte tensorflow-gpu balíček, který zahrnuje podporu GPU.

Ve virtuálním prostředí můžete použít příkaz pip místo pip3 a python místo python3 .

A je to! Úspěšně jste nainstalovali TensorFlow a můžete jej začít používat.

Pro ověření instalace spusťte následující příkaz, který vytiskne verzi TensorFlow:

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

V době psaní tohoto článku je nejnovější stabilní verze TensorFlow 2.2.0:

2.2.0

Vaše verze TensorFlow se může lišit od verze zobrazené zde.

Pokud jste v TensorFlow noví, navštivte stránku Začínáme s TensorFlow a zjistěte, jak vytvořit svou první aplikaci ML. Můžete také naklonovat repozitáře TensorFlow Modelsor TensorFlow-Examples z Github a prozkoumat a otestovat příklady TensorFlow.

Až budete s prací hotovi, deaktivujte prostředí zadáním deactivate a vrátíte se do svého normálního prostředí.

deactivate