Chcę używać Celery do uruchamiania zadań na serwerze GPU z czterema kartami Tesli. Prowadzę pracownika Selera z grupą czterech pracowników, dzięki czemu każda karta zawsze wykonuje jedną pracę.

Mój problem polega na tym, jak poinstruować pracowników, aby każdy żądał jednego GPU. Obecnie polegam na założeniu, że wszystkie procesy robocze powinny mieć ciągłe identyfikatory procesów:

device_id = os.getpid() % self.ndevices

Nie gwarantuję jednak, że zawsze będzie działać, tj. Gdy procesy robocze zostaną ponownie uruchomione z czasem. Idealnie więc chciałbym uzyskać identyfikator każdego pracownika bezpośrednio. Czy ktoś może mi powiedzieć, czy możliwe jest sprawdzenie pracownika w ramach zadania lub zasugerowanie innego rozwiązania w celu dystrybucji zadań na GPU?