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)

我懶得重裝,完美“解決”

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