머신러닝,딥러닝
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')
반응형