1. 程式人生 > 實用技巧 >ubuntu20.04安裝pytorch教程及其注意事項

ubuntu20.04安裝pytorch教程及其注意事項

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