首先貼上pytorch的官網:https://pytorch.org/
不多介紹,直接開始安裝步驟:
1、打開https://pytorch.org/get-started/locally/
2、選擇對應要安裝的版本。
第一行表示穩定版和預覽版
第二行表示系統
第三行選擇哪種安裝方式
第四行語言
第五行CUDA(看你的顯卡支不支持)如果沒有就選None。
如何查看支持的CUDA的版本,看這篇文章:https://blog.csdn.net/he99774/article/details/104713315
最後一行,打開cmd就可以安裝了。
在cmd安裝速度奇慢無比,只有幾kb每秒,800M的文件要下好幾天。
所以我是直接打開https://download.pytorch.org/whl/torch_stable.html
找到對應的版本,打開我的vpn,用瀏覽器下。
我是下的這兩個(自己下要選擇對應的python版本和cu還是cpu)
https://download.pytorch.org/whl/cu101/torch-1.4.0-cp35-cp35m-win_amd64.whl
https://download.pytorch.org/whl/cu101/torchvision-0.5.0-cp35-cp35m-win_amd64.whl
然後用pip安裝whl就好了。
國內的鏡像我沒找到,就這樣了。