Cómo instalar TensorFlow en Ubuntu 20.04

TensorFlow es una plataforma gratuita y de código abierto para el aprendizaje automático creada por Google. Lo utilizan varias organizaciones, incluidas Twitter, PayPal, Intel, Lenovo y Airbus.

TensorFlow se puede instalar en todo el sistema, en un entorno virtual de Python, como Dockercontainer o con Anaconda.

Este tutorial explica cómo instalar TensorFlow en un entorno virtual de Python en Ubuntu 20.04.

Un entorno virtual le permite tener varios entornos de Python aislados diferentes en una sola computadora e instalar una versión específica de un módulo por proyecto, sin preocuparse de que afecte a sus otros proyectos.

Instalando TensorFlow en Ubuntu 20.04 #

Ubuntu 20.04 viene con Python 3.8 por defecto. Puede verificar que Python 3 está instalado en su sistema escribiendo:

python3 -V

La salida debería verse así:

Python 3.8.2

La forma recomendada de crear un entorno virtual es mediante el venv módulo, que se incluye en el python3-venv paquete.

Para instalar python3-venv paquete ejecuta el siguiente comando:

sudo apt install python3-venv python3-dev

Una vez que el módulo esté instalado, estará listo para crear un entorno virtual para nuestro proyecto TensorFlow.

Navegue hasta el directorio donde le gustaría almacenar sus entornos virtuales de Python 3. Puede ser su directorio de inicio o cualquier otro directorio donde su usuario tenga permisos de lectura y escritura.

Crea un nuevo directorio para el proyecto TensorFlow y haz clic en él:

mkdir my_tensorflowcd my_tensorflow

Dentro del directorio, ejecute el siguiente comando para crear el entorno virtual:

python3 -m venv venv

El segundo venv es el nombre del entorno virtual. Puede utilizar cualquier nombre que desee para el entorno virtual.

El comando anterior crea un directorio llamado venv , que contiene una copia del binario de Python, el administrador de paquetes Pip, la biblioteca estándar de Python y otros archivos de soporte.

Para comenzar a usar el entorno virtual, actívelo ejecutando activate guión:

source venv/bin/activate

Una vez activado, el directorio bin del entorno virtual se agregará al comienzo del sistema $PATH variable. Además, el indicador del shell cambiará y mostrará el nombre del entorno virtual en el que se encuentra actualmente. En este ejemplo, es (venv) .

La instalación de TensorFlow requiere pip versión 19 o superior. Ingrese el siguiente comando para actualizar pip a la última versión:

pip install --upgrade pip

Ahora que el entorno virtual está activado, es hora de instalar el paquete TensorFlow.

pip install --upgrade tensorflow

Si tienes una GPU NVIDIA dedicada y quieres aprovechar su potencia de procesamiento, en lugar de tensorflow , instala el tensorflow-gpu paquete, que incluye compatibilidad con GPU.

Dentro del entorno virtual, puede usar el comando pip en lugar de pip3 y python en lugar de python3 .

¡Eso es todo! Ha instalado correctamente TensorFlow y puede comenzar a usarlo.

Para verificar la instalación, ejecute el siguiente comando, que imprimirá la versión de TensorFlow:

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

Al momento de escribir este artículo, la última versión estable de TensorFlow es 2.2.0:

2.2.0

Su versión de TensorFlow puede diferir de la versión que se muestra aquí.

Si es nuevo en TensorFlow, visite la página Primeros pasos con TensorFlow y aprenda cómo crear su primera aplicación ML. También puede clonar los repositorios TensorFlow Models o TensorFlow-Examples de Github y explorar y probar los ejemplos de TensorFlow.

Una vez que haya terminado con su trabajo, desactive el entorno escribiendo deactivate , y volverá a su shell normal.

deactivate