前言
在window上安裝pytorch雖然比caffe容易得多,但是也遇到了各種各樣得坑,這裏我把自己遇到的坑記錄分享一下。
DLL加載不成功的坑
起初,我的python版本是3.5,因爲網絡的原因,直接 pip install torch torchvision很難成功。 所以下載torch-1.0.0-cp35-cp35m-win_amd64.whl,手動安裝如下:
pip install torch-1.0.0-cp35-cp35m-win_amd64.whl
可是安裝成功後 去import torch時遇到了:
from torch._C import *
ImportError: DLL load failed: 找不到指定的模塊
網上有很多種解決方法,比如說 pip install intel-openmp或者添加python安裝目錄下torch的lib路徑到系統path,但我都試過了沒有用。
解決辦法
因爲我之前的python版本是3.5.1,所以我懷疑是因爲python版本低引起,所以又裝了python3.6,並且將其安裝路徑添加到系統路徑,確保在cmd下輸入pytho命令,其對應版本爲3.6。
然後在pytorch官網選取穩定版本:
將上面的command粘貼到cmd窗口:
安裝成功後,驗證ok:
結論
當遇到torch C dll無法加載的錯誤時,可以考慮將python版本升級來解決。