imagoseine : tensorflow et accès GPU

Bonjour,

J’ai réussi à installer un environnement Anaconda pour des entrainements de réseaux de neurones s’appuyant sur tensorflow.

J’utilise SLURM pour la première fois. Tout semble bien marcher de ce point de vue : j’ai lancé des tâches minimalistes en python. Mes commandes python renvoient les commandes attendues dans le fichier slurm mais la commande tensorflow permettant de lister les ressources disponibles :

tensorflow.config.list_physical_devices('GPU')

ne liste pas de GPU. Les outils nvidia (nvidia-smi…) ne sont pas accessibles (ce qui est sans doute nécessaire). Pouvez-vous m’indiquer une procédure pour accéder aux ressources GPU de cette manière ou toute autre démarche nécessaire ?

Cordialement

Bonjour,

Pour allouer un GPU vous devez rajouter cet argument à la liste des instructions :
--gres=gpu:1
ou, si vous voulez cibler une instance particulière :
--gres=gpu:a100_7g.80gb:1

Cordialement.

Bonjour, je reviens vers vous à propos de l’accès aux GPU via tensorflow. Comment puis-je faire tourner des scripts s’appuyant sur cette librairie ?

Bonjour,

J’ai déjà répondu à cette question ici : imagoseine : tensorflow et accès GPU - #2 by rey