这个报错是说Torch找不到cuda ,无法编译,要调用CPU环境。
方法:
第一步:
device = torch.device('cuda' if torch.cuda.is_available() else 'cpu')
#在代码开头加上这句
第二步:
然后使用快捷键ctrl+R将.cuda()改成.to(device)
这个报错是说Torch找不到cuda ,无法编译,要调用CPU环境。
方法:
第一步:
device = torch.device('cuda' if torch.cuda.is_available() else 'cpu')
#在代码开头加上这句
第二步:
然后使用快捷键ctrl+R将.cuda()改成.to(device)