Pytorch Illegal instruction 解决(鸵鸟战术)

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)

我懒得重装,完美“解决”

發表評論
所有評論
還沒有人評論,想成為第一個評論的人麼? 請在上方評論欄輸入並且點擊發布.
相關文章