ubuntu20.04安裝pytorch教程及其注意事項
阿新 • • 發佈:2020-11-28
Pytorch介紹
PyTorch是一個開源的Python機器學習庫,基於Torch,用於自然語言處理等應用程式。PyTorch既可以看作加入了GPU支援的numpy,同時也可以看成一個擁有自動求導功能的強大的深度神經網路。
它具有以下優勢:
PyTorch是相當簡潔且高效快速的框架
設計追求最少的封裝
設計符合人類思維,它讓使用者儘可能地專注於實現自己的想法
與google的Tensorflow類似,FAIR的支援足以確保PyTorch獲得持續的開發更新
PyTorch作者親自維護的論壇 供使用者交流和求教問題
入門簡單
…
至於其更廣泛更具體的優勢,參照知乎的討論:https://www.zhihu.com/question/65578911/answer/565574377
Pytorch安裝
pytorch的安裝難度要遠低於tensorflow。百度搜索並進入https://pytorch.org/get-started/locally/
下滑找到如下介面,即可選擇安裝方式:
選擇合適並版本複製Run this Command:即可進行安裝(不推薦)。
注意事項:安裝過程可能比較緩慢,可以用以下命令(推薦)
pip --default-timeout=1000 install --index-url https://mirrors.aliyun.com/pypi/simple torch==1.6.0+cpu torchvision==0.7.0+cpu -f https://download.pytorch.org/whl/torch_stable.html