Tengo acceso a través de ssh a un grupo de n GPU. Tensorflow les dio automáticamente nombres gpu: 0, ..., gpu: (n-1).

Otros también tienen acceso y a veces toman gpus aleatorios. No puse ningunatf.device() explícitamente porque eso es engorroso e incluso si seleccioné el número de gpu j y que alguien ya está en el número de gpu j sería problemático.

Me gustaría ir a través del uso de gpus y encontrar el primero que no se usa y usar solo este. Supongo que alguien podría analizar la salida denvidia-smi con bash y obtenga una variable i y alimente esa variable i a la secuencia de comandos de tensorflow como el número de gpu a usar.

Nunca he visto ningún ejemplo de esto. Me imagino que es un problema bastante común. ¿Cuál sería la forma más sencilla de hacer eso? ¿Está disponible un tensorflow puro?