PyTorch 的安裝可以到官網 https://pytorch.org/，選擇適合自己機器以及安裝方式，執行對應的命令即可。

wohu@ubuntu1804:~$ sudo pip3  install torch torchvision

如果下載速度太慢，請使用清華源安裝

sudo pip3  install -i https://pypi.tuna.tsinghua.edu.cn/simple  torch torchvision

有以下輸出，說明安裝成功

In [1]: import torch as t                                                       In [2]: t.__version__                                                           
Out[2]: '1.5.1'In [3]:    

除了安裝 PyTorch 之外，建議也安裝 torchvision 包。torchvision 包是服務于 PyTorch 深度學習框架的，用來生成圖片、視頻數據集和一些流行的模型類和預訓練模型。

簡單來說，torchvision 由以下四個模塊組成。

• torchvision.datasets
• torchvision.models
• torchvision.transforms
• torchvision.utils

總結

以上是生活随笔為你收集整理的PyTorch 笔记（01）— Ubuntu 使用 pip 清华源安装 PyTorch的全部內容，希望文章能夠幫你解決所遇到的問題。

如果覺得生活随笔網站內容還不錯，歡迎將生活随笔推薦給好友。