这个报错是说Torch找不到cuda ,无法编译,要调用CPU环境。

方法:
第一步:

device = torch.device('cuda' if torch.cuda.is_available() else 'cpu')
#在代码开头加上这句

第二步:
然后使用快捷键ctrl+R将.cuda()改成.to(device)

Logo

有“AI”的1024 = 2048,欢迎大家加入2048 AI社区

更多推荐