Pytorch - Illegal instruction 解决
- 运行环境为 anaconda3 下的虚拟环境: Python3 + cuda9.0 + Pytorch1.1.0
运行代码利用网络 net 对输入数据 x 求解时,即运行 net(x) 时出现 Illegal instruction (core dumped)
利用 anaconda 虚拟环境 python2.7 及其 Pytorch 运行相同代码,正确运行,表明当前环境有问题
查找网上解决方案,一句话,CPU 不兼容,需要重新安装 Ptorch
我太懒了,我真的不想装
偷懒方案:将运行迁移至 GPU
device = torch.device('cuda')
x = x.to(device)
net = net.to(device)