Comment installer TensorFlow sous Ubuntu 16.04 avec Virtualenv ?

Le guide ci-dessous explique comment installer TensorFlow sur Ubuntu 16.04 avec virtualenv:

1. Installation



Étant donné que Python 3 est installé nativement sous Ubuntu 16.04, l’installation de TensorFlow peut-être faite rapidement avec les instructions suivantes:

$ sudo apt-get install python3-pip python3-dev python-virtualenv
$ virtualenv --system-site-packages -p python3 ~/tensorflow
$ source ~/tensorflow/bin/activate
(tensorflow)$ pip3 install --upgrade tensorflow

Il est à noter que l’installation se fait dans un environnement virtuel, ce qui permet de l’isoler des autres développement Python.

2. Tester l’installation

Créez le fichier tf_test.py contenant le script suivant :

# Python
import tensorflow as tf
hello = tf.constant('Hello, TensorFlow!')
sess = tf.Session()
print(sess.run(hello))

Dans l’environnement virtuel TensorFlow, exécuter la commande suivante:

(tensorflow)$ python3 tf_test.py 

Ce script devrait afficher:

'Hello, TensorFlow!'

3. Environnement virtuel

Activation : la commande ci-dessous active l’environnement virtuel.

$ source ~/tensorflow/bin/activate

Désactivation : Pour désactiver l’environnement virtuel TensorFlow.

(tensorflow)$ deactivate