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)