05/08/2020
TensorFlow es una de las librerías más populares para el aprendizaje automático ( Machine Learning ) y la inteligencia artificial ( IA ) en Python. Su versatilidad y potencia la han convertido en una herramienta esencial para desarrolladores e investigadores en el campo.
Este artículo cubre todo lo que necesitas saber sobre la instalación, configuración y uso de TensorFlow con Python, incluyendo diferentes escenarios y resolución de problemas comunes.
Instalación de TensorFlow
Requisitos del Sistema
Antes de instalar TensorFlow, asegúrate de cumplir los siguientes requisitos:
- Sistema operativo: Ubuntu 104 o superior (64 bits) es oficialmente compatible. Otras distribuciones de Linux pueden funcionar, pero no están garantizadas.
- Python : Se recomienda una versión reciente de Python (7 o superior).
- (Opcional) GPU compatible con CUDA y controladores NVIDIA instalados si deseas aprovechar el procesamiento de GPU.
Entornos Virtuales
Se recomienda encarecidamente utilizar entornos virtuales para gestionar las dependencias de tus proyectos. Esto evita conflictos entre diferentes proyectos y mantiene la limpieza de tu sistema. Puedes crear un entorno virtual con venv:
python3 -m venv tfsource tf/bin/activate
Instalación con pip
La forma más sencilla de instalar TensorFlow es usando pip, el gestor de paquetes de Python. Actualiza pipprimero:

pip install --upgrade pip
Luego, instala TensorFlow :
- Para CPU:
pip install tensorflow - Para GPU:
pip install tensorflow-gpu(requiere CUDA y CuDNN)
Nota importante: Evita usar condapara la instalación de TensorFlow, ya que puede no incluir la versión más reciente y estable. pipes el método oficial para instalar desde PyPI.
Instalación en ARM (Aarch64/ARM64)
Para procesadores Aarch64/ARM64, las compilaciones de CPU de Linux son proporcionadas por un tercero (AWS). Al instalar el paquete tensorflowen una máquina ARM, se instala el paquete tensorflow-cpu-awsde AWS.
Verificación de la Instalación
Después de la instalación, verifica que TensorFlow esté funcionando correctamente. Para CPU:
python3 -c "import tensorflow as tf; print(tf.reduce_sum(tf.random.normal([1000, 1000])))"
Para GPU:

python3 -c "import tensorflow as tf; print(tf.config.list_physical_devices('GPU'))"
Si la salida muestra un tensor (CPU) o una lista de dispositivos GPU (GPU), la instalación fue exitosa.
Configuración de GPU (Solo GPU)
Si la verificación de GPU falla, es posible que haya un conflicto con la instalación de CUDA existente. Debes crear enlaces simbólicos:
pushd $(dirname $(python -c 'print(__import__("tensorflow").__file__)'))ln -svf ../nvidia//lib/.so .popdln -sf $(find $(dirname $(dirname $(python -c "import nvidia.cuda_nvcc; print(nvidia.cuda_nvcc.__file__)")))//bin/) -name ptxas -print -quit ) $VIRTUAL_ENV/bin/ptxas
Después de esto, vuelve a ejecutar la verificación de GPU.
Versiones de Python para TensorFlow
Aunque TensorFlow admite varias versiones de Python, se recomienda utilizar la versión más reciente y estable disponible para asegurar la mejor compatibilidad y acceso a las últimas características y correcciones de errores. Siempre consulta la documentación oficial de TensorFlow para conocer las versiones de Python oficialmente compatibles.
TensorFlow y Python: Un Ejemplo Práctico
A continuación, se muestra un ejemplo simple de cómo usar TensorFlow con Python para construir un modelo de clasificación de imágenes usando el conjunto de datos MNIST:
# Importa TensorFlowimport tensorflow as tf# Carga y prepara el conjunto de datos MNISTmnist = tf.keras.datasets.mnist(x_train, y_train), (x_test, y_test) = mnist.load_data()x_train, x_test = x_train / 250, x_test / 250# Construye el modelomodel = tf.keras.models.Sequential([ tf.keras.layers.Flatten(input_shape=(28, 28)), tf.keras.layers.Dense(128, activation='relu'), tf.keras.layers.Dropout(0.2), tf.keras.layers.Dense(10, activation=oftmax')])# Compila el modelomodel.compile(optimizer='adam', loss=parse_categorical_crossentropy', metrics=['accuracy'])# Entrena el modelomodel.fit(x_train, y_train, epochs=5)# Evalúa el modelomodel.evaluate(x_test, y_test, verbose=2)Este código muestra un ejemplo básico de cómo construir, entrenar y evaluar un modelo de aprendizaje automático simple con TensorFlow. La estructura de un modelo de TensorFlow se basa en capas. En este ejemplo, se utiliza un modelo secuencial con capas densas y una capa de abandono (dropout) para la regularización.
Tabla Comparativa de Versiones de TensorFlow y Python
A continuación, una tabla que resume las compatibilidades, aunque es importante siempre revisar la documentación oficial.
| Versión TensorFlow | Versiones Python Compatibles |
|---|---|
| 10 | 7, 8, 9, 10, 11 |
| 9 | 7, 8, 9 |
| 8 | 7, 8, 9 |
Nota: Esta tabla es solo una referencia. Siempre consulta la documentación oficial de TensorFlow para la información más actualizada sobre compatibilidad de versiones.
Solución de Problemas Comunes
A continuación se presentan algunos problemas comunes durante la instalación y uso de TensorFlow y sus posibles soluciones:
Error: No module named ‘tensorflow’
Solución: Verifica que TensorFlow esté instalado en el entorno virtual correcto y que el entorno esté activado.
Error: DLL Load Failed
Solución: (Windows) Instala las bibliotecas redistribuibles de Microsoft Visual C++.
Errores de GPU
Solución: Asegúrate de tener los controladores de GPU y CUDA/CuDNN correctos instalados y compatibles con tu tarjeta gráfica.
TensorFlow es una librería poderosa y versátil para el aprendizaje automático. Con una correcta instalación y comprensión de sus fundamentos, puedes empezar a desarrollar modelos complejos e innovadores. Recuerda siempre consultar la documentación oficial para obtener información actualizada y detallada.
Si quieres conocer otros artículos parecidos a Librería tensorflow con python puedes visitar la categoría Libros y Librerías.
