0
llucycodes42
Library: pytorch
pytorch
Shortcut: pytorch.device
pytorch.device
0 Comments
device = torch.device('cuda:0' if torch.cuda.is_available() else 'cpu')