Para rodar o tensorflow é possível instalar com o conda ou rodar direto via apptainer. O conda roda fácil tensorflow até a versão 2.4. Recomendamos usar o apptainer.

apptainer pull docker://tensorflow/tensorflow:latest-gpu
apptainer run --nv tensorflow_latest-gpu.sif