5 lines
86 B
Python
5 lines
86 B
Python
import torch
|
|
|
|
def device():
|
|
return 'cuda' if torch.cuda.is_available() else 'cpu'
|