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