Tensorflow es una biblioteca de código abierto para computación numérica, Machine Learning y Deep Learning.

Se sugiere el trabajo con Tensorflow en entornos virtuales:

1. Crear un entorno virtual con Anaconda, por ejemplo tf_gpu

2. Activar el entorno virtual

source ~/.bashrc
conda activate tf_gpu

3. Instalar en el entorno virtual las bibliotecas necesarias

conda install tensorflow-gpu=2.2.0 Keras=2.4.3 cudatoolkit=10.1.243 cudnn=7.6.5.

4. Verificar el funcionamiento de Tensorflow. Para ello, se puede desarrollar un código en Python con el siguiente contenido y enviarlo a ejecutar con el comando sbatch:

import tensorflow as tf
print("Tensorflow is built with cuda = ", tf.test.is_built_with_cuda())
print("Device name = ", tf.test.gpu_device_name())