Pytorch 安裝及中文文件
windows下安裝PyTorch0.3.0
首先先宣告:希望把python版本設定為3.5.因為在windows上安裝深度學習框架,無論是TensorFlow還是Pytorch,都不支援python2,由於TF不支援3.6,所以我推薦大家使用3.5.不過3.6的安裝方法也會給大家,大家也可以試試。
windows版本要求64位。32位的同學。。。可以更新了。已經2018年了。32位系統不能儲存單個檔案大於4G的。。
然後,大家pip應該有吧,這個教程很簡單,不需要conda也不需要別的亂七八糟的,就最基礎的,pip就行。
然後去網盤下載:
連結:https://pan.baidu.com/s/1dUrI3O 密碼:lahq
有兩個版本一個對應3.5一個對應3.6.然後下載以後,在這個路徑下 pip install "torch-0.3.0b0+591e73e-cp35-cp35m-win_amd64.whl"(輸入pip install 然後點選一下或多下tab,自動會有這個檔案出現)
然後就裝完了。簡單嗎。測試一下:
如果沒有GPU,那其實安裝已經結束了,但我相信大部分人,都是要gpu版本的。
去安裝一下cuda8+cudnn6.或者cuda9+cudnn7。我是前者,因為我裝過TensorFlow只支援cuda8。網上教程很多,不過坑,其實只有一個。安裝cuda請選擇自定義安裝,只勾選cuda,其他一律不選!
如果你在安裝pytorch之前沒有裝cuda,那麼你需要先
- pip uninstall torch
然後再重新裝一遍,測試的話:
然後,你們還可以試試具體案例(官方demo有的跑不起來,需要略作修改,可能是因為PyTorch版本更新太快的原因),比如我寫的一個訓練好的ResNet:我的github,直接下載到一個資料夾下,
- python load_resnet.py
然後它會下載訓練集,給出結果。
轉自:https://blog.csdn.net/qj30212/article/details/79072517
Pytorch中文文件已經發布,完美翻譯,更加方便大家瀏覽:
Pytorch中文網:https://ptorch.com/
Pytorch中文文件:https://ptorch.com/docs/1/
Pytorch中文論壇:https://discuss.ptorch.com/
莫煩Ptorch : https://morvanzhou.github.io/tutorials/machine-learning/torch/