pytorch - 调用 cpu - Githubissues

ooolky / NoteBook

0 stars 0 forks source link

pytorch - 调用 cpu #39

Open ooolky opened 11 months ago

ooolky commented 11 months ago

操作：device = torch.device('cuda' if torch.cuda.is_available() else 'cpu')
参考