머신러닝,딥러닝

device cuda available code

mcdn 2023. 1. 27. 15:55
반응형
device = torch.device('cuda' if torch.cuda.is_available() else 'cpu')

device = torch.device('cuda' if torch.cuda.is_available() else 'cpu')

반응형