Spaces:
Sleeping
Sleeping
File size: 141 Bytes
6e9c433 |
1 2 3 4 5 6 |
import torch
cpu_device = torch.device("cpu")
# device = torch.device("cuda:3" if torch.cuda.is_available() else "cpu")
device = cpu_device
|