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
 安裝好后進行測試:

 輸入以上命令即可完成測試是否成功安裝pytorch以及pytorch版本。

 


免責聲明!

本站轉載的文章為個人學習借鑒使用,本站對版權不負任何法律責任。如果侵犯了您的隱私權益,請聯系本站郵箱yoyou2525@163.com刪除。



 
粵ICP備18138465號   © 2018-2025 CODEPRJ.COM