Cómo instalar la GPU TensorFlow en Ubuntu 18.04

Introducción

TensorFlow es una plataforma de aprendizaje automático de código abierto. Google diseñó el software para ayudar a los investigadores, científicos de datos y desarrolladores a simplificar el proceso de implementación de modelos de aprendizaje automático.

Esta biblioteca integral para computación numérica puede ejecutarse en múltiples CPU, GPU y sistemas operativos móviles. En este tutorial, aprenda cómo instalar TensorFlow en Ubuntu 18.04.

Requisitos previos

• Un sistema Ubuntu Linux (16.04 o posterior)
• Acceso a una línea de comando/ventana de terminal (Ctrl +Alt +T )
• Una cuenta de usuario con sudo privilegios
• Pip 19.0 o posterior

Paso 1:Instale los paquetes necesarios

Antes de poder instalar TensorFlow, deberá configurar el entorno de desarrollo de Python. Incluye el siguiente software:

• Pitón (versión 3.4 o posterior)
• el administrador de paquetes pip (no anterior a la versión 19.0)
• Entorno virtual (software para aislar entornos Python)

Abra la ventana de la terminal y comience actualizando el repositorio con:

sudo apt update

A continuación, instale los dos primeros paquetes con el comando:

sudo apt install python3-dev python3-pip

Luego, ejecute el siguiente comando para una instalación de sistema amplio de Virtualenv:

sudo pip3 install -U virtualenv

Paso 2:Creación de un entorno virtual

Ahora que tiene Virtualenv en su sistema Ubuntu, puede usarlo para crear y aislar entornos de Python.

Crea tu primer entorno en un nuevo ./venv directorio:

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

Luego, active el entorno virtual para comenzar a trabajar dentro de él. Ejecute el siguiente comando:

source ./venv/bin/activate

Su indicador de shell ahora debería tener un (venv) prefijo como en la imagen de abajo:

Una vez que active venv , continúe con la instalación de pip dentro del entorno aislado:

pip install --upgrade pip

Si desea ver una lista completa de todos los paquetes dentro del entorno virtual, use el comando:

pip list

Muestra todos los paquetes y sus respectivas versiones, como en la siguiente imagen:

Paso 3:Instalación de TensorFlow

El siguiente paso difiere dependiendo de si está instalando TensorFlow para CPU o TensorFlow para GPU . La elección depende de la naturaleza de su carga de trabajo y opciones de hardware.

Opción 1:Instalar TensorFlow para CPU

El TensorFlow predeterminado paquete de software admite cargas de trabajo basadas en CPU. Para instalar el paquete y sus dependencias, escriba el siguiente comando:

pip install --upgrade tensorflow

Opción 2:Instalar TensorFlow para GPU

TensorFlow para GPU requiere una GPU habilitada para NVIDIA CUDA® dedicada y controladores relacionados. Este paquete de software está diseñado para cargas de trabajo de aprendizaje automático basadas en GPU.

Para instalar la última versión estable de TensorFlow para GPU , ejecute el comando:

pip install --upgrade tensorflow-gpu

Para obtener una lista de los requisitos de hardware y software para TensorFlow para GPU, consulte la documentación de TensorFlow sobre compatibilidad con GPU.

Opción 3:instalar versiones antiguas de TensorFlow

Versiones anteriores de TensorFlow para CPU y GPU también están disponibles para descargar.

Versión 1.14 y anteriores se instala ejecutando el comando en el siguiente formato:

pip install tensorflow==package_version

Para instalar TensorFlow para CPU 1.14 , ejecute el comando:

pip install tensorflow==1.14

Para instalar TensorFlow para GPU 1.14 , ejecute el comando:

pip install tensorflow-gpu==1.14

Versión 1.15 admite cargas de trabajo basadas en CPU y GPU. Para instalar TensorFlow 1.15 , escriba el comando:

pip install tensorflow-gpu==1.15rc2

Paso 4:Verificación de la instalación de TensorFlow

Para verificar la instalación de TensorFlow en Ubuntu, ingrese el comando en la ventana del terminal:

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