Guide d'installation de TensorFlow sur Ubuntu

Si vous aimez l'apprentissage automatique, vous devrez peut-être utiliser TensorFlow, sinon PyTorch.

En supposant que vous utilisez Linux, je vais vous montrer comment installer Tensorflow sur Ubuntu. J'ai utilisé Ubuntu 18.04 mais les étapes devraient également être valables pour les autres versions.

Notez que pendant que vous pouvez essayer le construire à partir de la source pour les autres plates-formes/distributions, TensorFlow prend principalement en charge Ubuntu Linux. Je vais donc m'en tenir à cela dans cet article.

Installation de TensorFlow sur Ubuntu Linux

Avant de commencer à configurer TensorFlow, vous devez activer le référentiel Universe sur Ubuntu. Vous pouvez le faire en utilisant cette commande :

sudo add-apt-repository universe

Ou accédez à Logiciels et mises à jour options et activez-le à partir de là :

Étape 1 :Obtenir l'environnement de développement Python

Tout d'abord, vous devez configurer un environnement de développement Python pour continuer. Par défaut, Python 3.6.x doit être installé sur Ubuntu. Vous pouvez vérifier la version de Python en utilisant :

python3 --version

Assurez-vous d'avoir Python 3.5–3.7 (comme indiqué officiellement). Si vous ne l'avez pas ou si vous devez le mettre à niveau, installez-le à l'aide de la commande suivante dans le terminal :

sudo apt install python3-dev

Étape 2 :Obtenir un pip

Ensuite, vous devez vérifier si vous avez le gestionnaire de paquets pip de Python.

pip3 --version

Si vous ne l'avez pas installé, vous pouvez l'installer en tapant ceci :

sudo apt install python3-pip

Cependant, il se peut que vous n'ayez pas la dernière version à bord.

Remarque : TensorFlow nécessite pip version 19.0 ou supérieure . Ne vous inquiétez pas, vous pouvez mettre à niveau pip une fois que vous avez configuré l'environnement virtuel dans la prochaine série d'étapes.

Étape 3 :Configurer l'environnement virtuel Python

Maintenant que vous avez terminé la configuration de Python et de pip, vous devez configurer un environnement virtuel pour le développement Python. De cette façon, vous n'affecterez pas le système hôte avec ce que vous faites avec la configuration de Python.

Pour commencer, installez virtualenv :

sudo -H pip3 install -U virtualenv

Le drapeau -H définit la variable d'environnement HOME sur le répertoire personnel.

Vous voudriez créer l'environnement virtuel Python en spécifiant un répertoire et en choisissant un interpréteur Python. Voici ce que vous devez taper pour ce faire :

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

Une fois cela terminé, il vous suffit d'activer l'environnement virtuel, pour cela, utilisez la commande source de la manière suivante :

source ./venv/bin/activate

Pendant que votre environnement virtuel est actif, vous observerez que votre invite de shell sera préfixée par (venv) comme indiqué dans l'image ci-dessous.

Ensuite, vous devez mettre à niveau la version de pip dans l'environnement virtuel sans affecter la configuration de l'hôte (si elle est déjà à jour, vous en serez informé ):

pip install --upgrade pip

Si vous êtes curieux, vous pouvez également consulter la liste des packages installés dans l'environnement virtuel en tapant ceci :

pip list

Étape 4 :Installer TensorFlow

Enfin, vous êtes sur le point d'installer TensorFlow.

Assurez-vous que vous utilisez l'environnement virtuel. Saisissez la commande suivante pour installer TensorFlow :

pip install --upgrade tensorflow

Vous pouvez également choisir de vérifier l'installation en saisissant la commande suivante dans l'environnement virtuel :

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

Enfin, vous avez installé TensorFlow avec succès sur Ubuntu !

Installation de TensorFlow sur Pop !_OS

Contrairement à Ubuntu, si vous avez Pop!_OS, vous n'avez pas besoin de suivre toutes ces étapes mais une seule commande pour utiliser votre système de base python.

Il vous suffit de saisir la commande suivante dans le terminal :

sudo apt install tensorflow-cuda-latest

Pour plus d'informations sur son utilisation, vous pouvez vous référer aux ressources officielles de Pop!_OS.

Conclusion

Gardez à l'esprit que chaque fois que vous souhaitez utiliser TensorFlow sur Ubuntu, vous devez d'abord entrer dans l'environnement virtuel Python spécifié.

Si vous débutez avec TensorFlow, vous voudrez peut-être consulter les ressources officielles disponibles pour apprendre.

J'espère que cela vous aidera. Il convient également de noter que j'ai effectué les étapes pour installer TensorFlow sur une nouvelle installation d'Ubuntu - donc si vous avez modifié quelque chose sur votre installation, vous devrez peut-être résoudre un peu le problème lors de l'installation de TensorFlow sur Ubuntu.

N'hésitez pas à me faire part de vos réflexions dans les commentaires ci-dessous.