Jak nainstalovat TensorFlow GPU na Ubuntu 18.04

Úvod

TensorFlow je open source platforma pro strojové učení. Google navrhl software, aby pomohl výzkumníkům, datovým vědcům a vývojářům zjednodušit proces implementace modelů strojového učení.

Tato komplexní knihovna pro numerické výpočty může běžet na více CPU, GPU i mobilních operačních systémech. V tomto tutoriálu se dozvíte, jak nainstalovat TensorFlow na Ubuntu 18.04.

Předpoklady

• Systém Ubuntu Linux (16.04 nebo novější)
• Přístup do příkazového řádku/okna terminálu (Ctrl +Alt +T )
• Uživatelský účet s sudo privilegia
• Pip 19.0 nebo novější

Krok 1:Nainstalujte požadované balíčky

Než budete moci nainstalovat TensorFlow, budete muset nastavit vývojové prostředí Python. Zahrnuje následující software:

• Python (verze 3.4 nebo novější)
• správce balíčků pip (ne starší než verze 19.0)
• Virtuální (software pro izolaci prostředí Pythonu)

Otevřete okno terminálu a začněte aktualizací úložiště pomocí:

sudo apt update

Dále nainstalujte první dva balíčky pomocí příkazu:

sudo apt install python3-dev python3-pip

Poté spusťte následující příkaz pro instalaci Virtualenv pro celý systém:

sudo pip3 install -U virtualenv

Krok 2:Vytvoření virtuálního prostředí

Nyní, když máte ve svém systému Ubuntu Virtualenv, můžete jej použít k vytváření a izolaci prostředí Python.

Vytvořte své první prostředí v novém ./venv adresář:

virtualenv --system-site-packages -p python3 ./venv

Poté aktivujte virtuální prostředí a začněte v něm pracovat. Spusťte následující příkaz:

source ./venv/bin/activate

Vaše výzva shellu by nyní měla mít (venv) prefix jako na obrázku níže:

Jakmile aktivujete venv , přejděte k instalaci pip uvnitř izolovaného prostředí:

pip install --upgrade pip

Pokud chcete vidět úplný seznam všech balíčků ve virtuálním prostředí, použijte příkaz:

pip list

Zobrazí všechny balíčky a jejich příslušné verze, jako na následujícím obrázku:

Krok 3:Instalace TensorFlow

Další krok se liší v závislosti na tom, zda instalujete TensorFlow pro CPU nebo TensorFlow pro GPU . Výběr závisí na povaze vaší pracovní zátěže a možnostech hardwaru.

Možnost 1:Nainstalujte TensorFlow pro CPU

Výchozí TensorFlow softwarový balík podporuje zátěže založené na CPU. Chcete-li nainstalovat balíček a jeho závislosti, zadejte následující příkaz:

pip install --upgrade tensorflow

Možnost 2:Nainstalujte TensorFlow pro GPU

TensorFlow pro GPU vyžaduje vyhrazený GPU s podporou NVIDIA CUDA® a související ovladače. Tento softwarový balíček je určen pro úlohy strojového učení založené na GPU.

Chcete-li nainstalovat nejnovější stabilní verzi TensorFlow pro GPU , spusťte příkaz:

pip install --upgrade tensorflow-gpu

Seznam hardwarových a softwarových požadavků pro TensorFlow pro GPU najdete v dokumentaci TensorFlow o podpoře GPU.

Možnost 3:Instalace starých verzí TensorFlow

Starší verze TensorFlow pro CPU a GPU jsou také k dispozici ke stažení.

Verze 1.14 a starší se nainstaluje spuštěním příkazu v následujícím formátu:

pip install tensorflow==package_version

Chcete-li nainstalovatTensorFlow pro CPU 1.14 , spusťte příkaz:

pip install tensorflow==1.14

Chcete-li nainstalovat TensorFlow pro GPU 1.14 , spusťte příkaz:

pip install tensorflow-gpu==1.14

Verze 1.15 podporuje pracovní zátěže založené na CPU i GPU. Chcete-li nainstalovat TensorFlow 1.15 , zadejte příkaz:

pip install tensorflow-gpu==1.15rc2

Krok 4:Ověření instalace TensorFlow

Chcete-li ověřit instalaci TensorFlow v Ubuntu, zadejte příkaz v okně terminálu:

python -c "import tensorflow as tf;print(tf.reduce_sum(tf.random.normal([1000, 1000])))"