Come installare la GPU TensorFlow su Ubuntu 18.04

Introduzione

Flusso tensore è una piattaforma di apprendimento automatico open source. Google ha progettato il software per aiutare ricercatori, data scientist e sviluppatori a semplificare il processo di implementazione dei modelli di apprendimento automatico.

Questa libreria end-to-end per il calcolo numerico può essere eseguita su più CPU, GPU e sistemi operativi mobili. In questo tutorial, scopri come installare TensorFlow su Ubuntu 18.04.

Prerequisiti

• Un sistema Ubuntu Linux (16.04 o successivo)
• Accesso a una riga di comando/finestra del terminale (Ctrl +Alt +T )
• Un account utente con sudo privilegi
• Pip 19.0 o successivo

Passaggio 1:installa i pacchetti richiesti

Prima di poter installare TensorFlow, dovrai configurare l'ambiente di sviluppo Python. Include il seguente software:

• Pitone (versione 3.4 o successive)
• il gestore di pacchetti pip (non precedente alla versione 19.0)
• Virtualv (software per isolare ambienti Python)

Apri la finestra del terminale e inizia aggiornando il repository con:

sudo apt update

Quindi, installa i primi due pacchetti con il comando:

sudo apt install python3-dev python3-pip

Quindi, esegui il comando seguente per un'installazione su tutto il sistema di Virtualenv:

sudo pip3 install -U virtualenv

Fase 2:creazione di un ambiente virtuale

Ora che hai Virtualenv sul tuo sistema Ubuntu, puoi usarlo per creare e isolare ambienti Python.

Crea il tuo primo ambiente in un nuovo ./venv directory:

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

Quindi, attiva l'ambiente virtuale per iniziare a lavorare al suo interno. Esegui il seguente comando:

source ./venv/bin/activate

Il prompt della shell ora dovrebbe avere un (venv) prefisso come nell'immagine qui sotto:

Dopo aver attivato venv , passa all'installazione di pip all'interno dell'ambiente isolato:

pip install --upgrade pip

Se vuoi vedere un elenco completo di tutti i pacchetti all'interno dell'ambiente virtuale, usa il comando:

pip list

Visualizza tutti i pacchetti e le rispettive versioni, come nell'immagine seguente:

Fase 3:installazione di TensorFlow

Il passaggio successivo varia a seconda che tu stia installando TensorFlow per CPU o TensorFlow per GPU . La scelta dipende dalla natura del carico di lavoro e dalle opzioni hardware.

Opzione 1:installa TensorFlow per CPU

Il TensorFlow predefinito pacchetto software supporta carichi di lavoro basati su CPU. Per installare il pacchetto e le sue dipendenze, digita il seguente comando:

pip install --upgrade tensorflow

Opzione 2:installa TensorFlow per GPU

TensorFlow per GPU richiede una GPU dedicata abilitata per NVIDIA CUDA® e relativi driver. Questo pacchetto software è pensato per carichi di lavoro di apprendimento automatico basati su GPU.

Per installare l'ultima versione stabile di TensorFlow per GPU , esegui il comando:

pip install --upgrade tensorflow-gpu

Per un elenco dei requisiti hardware e software per TensorFlow per GPU, fare riferimento alla documentazione di TensorFlow sul supporto GPU.

Opzione 3:installazione di vecchie versioni di TensorFlow

Versioni precedenti di TensorFlow per CPU e GPU sono anche disponibili per il download.

Versione 1.14 e precedenti viene installato eseguendo il comando nel seguente formato:

pip install tensorflow==package_version

Per installare TensorFlow per CPU 1.14 , esegui il comando:

pip install tensorflow==1.14

Per installare TensorFlow per GPU 1.14 , esegui il comando:

pip install tensorflow-gpu==1.14

Versione 1.15 supporta carichi di lavoro basati su CPU e GPU. Per installare TensorFlow 1.15 , digita il comando:

pip install tensorflow-gpu==1.15rc2

Fase 4:verifica dell'installazione di TensorFlow

Per verificare l'installazione di TensorFlow in Ubuntu, inserisci il comando nella finestra del terminale:

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