device_id.py 176 B

123456789101112
  1. from enum import IntEnum
  2. class DeviceId(IntEnum):
  3. GPU0 = 0,
  4. GPU1 = 1,
  5. GPU2 = 2,
  6. GPU3 = 3,
  7. GPU4 = 4,
  8. GPU5 = 5,
  9. GPU6 = 6,
  10. GPU7 = 7,
  11. CPU = 99