Spaces:
Runtime error
Runtime error
File size: 176 Bytes
47c60f9 |
1 2 3 4 5 6 7 8 9 10 11 12 13 |
from enum import IntEnum
class DeviceId(IntEnum):
GPU0 = 0,
GPU1 = 1,
GPU2 = 2,
GPU3 = 3,
GPU4 = 4,
GPU5 = 5,
GPU6 = 6,
GPU7 = 7,
CPU = 99
|