當前位置:
首頁 > 最新 > Windows下安裝PyTorch0.4.0

Windows下安裝PyTorch0.4.0

2018年4月25號,PyTorch 官方發布 0.4.0 版本,該版本的 PyTorch 有多項重大更新,其中最重要的改進是官方支持 Windows (詳細改動見Pytorch官方GitHub):Windows各種安裝方式匯總:

Windows用戶能直接通過conda、pip和源碼編譯三種方式來安裝Pytorch了,假如我們用pip來安裝的話:

安裝GPU版本,如:Windows+pip+python 3.6+CUDA 8 ,只需在cmd中輸入:

如果電腦沒有英偉達顯卡,那就只能安裝CPU版本的,如:Windows+pip+python 3.6 ,在cmd中輸入:

最後需要安裝 torchvision,我是通過本地安裝的方式進行安裝的,下載地址:(https://pypi.python.org/pypi/torchvision/0.2.1)下載好之後,用cmd進入torchvision所在的文件夾下,輸入下面的命令安裝:

測試一下是否安裝好了,看看能不能用GPU加速:

安裝後會出現一些問題,官方文檔針對一些問題給出了解決方案,在知乎上有大神也給出了一些解決方案。

喜歡這篇文章嗎?立刻分享出去讓更多人知道吧!

本站內容充實豐富,博大精深,小編精選每日熱門資訊,隨時更新,點擊「搶先收到最新資訊」瀏覽吧!


請您繼續閱讀更多來自 StudyNLP 的精彩文章:

TAG:StudyNLP |