共計 2381 個字符，預計需要花費 6 分鐘才能閱讀完成。
本篇內容主要講解“Linux 下怎么安裝 PyTorch”，感興趣的朋友不妨來看看。本文介紹的方法操作簡單快捷，實用性強。下面就讓丸趣 TV 小編來帶大家學習“Linux 下怎么安裝 PyTorch”吧!
一、PyTorch 簡介
PyTorch 是一個開源的 Python 機器學習庫，基于 Torch，用于自然語言處理等應用程序。2017 年 1 月，由 Facebook 人工智能研究院（FAIR）基于 Torch 推出 PyTorch。PyTorch 的前身是 Torch，其底層和 Torch 框架一樣，但是使用 Python 重新寫了很多內容，不僅更加靈活，支持動態圖，而且提供了 Python 接口。它是由 Torch7 團隊開發，是一個以 Python 優先的深度學習框架，不僅能夠實現強大的 GPU 加速，同時還支持動態神經網絡。PyTorch 既可以看作加入了 GPU 支持的 numpy，同時也可以看成一個擁有自動求導功能的強大的深度神經網絡。除了 Facebook 外，它已經被 Twitter、CMU 和 Salesforce 等機構采用。
二、安裝步驟 1、操作系統選擇
emsp; emsp; 查看 PyTorch 官網，可以看到 PyTorch 支持 Linux、Mac、window 平臺、支持 conda、pip、源碼等安裝方式，也支持 CPU、cuda、ROCm 計算平臺，我們點擊環境選擇可以發現目前只有 linux 系統是支持全語言、全安裝方式、全計算平臺的，所以我們選擇 linux 操作系統作為系統環境。另外機器學習計算要求 glibc 版本要求較高，centos 搭載的內核和 glbic 版本較低，Ubuntu 搭載的內核版本都較新，所以機器學習主機建議使用 Ubuntu 操作系統。目前 cuda 更新支持的最低 Ubuntu 版本為 18.04，所以建議使用 Ubuntu18.04 以上的操作系統。
wuhs@s169:~$ cat /etc/os-release
NAME=“Ubuntu”
VERSION=“18.04.6 LTS (Bionic Beaver)”
2、Anaconda3 安裝
emsp; emsp; 如上所示，PyTorch 支持的安裝方式有多種，博主擬采用 conda 安裝方式，建議先安裝 Anaconda3，可以根據我們需要創建不同虛擬環境，虛擬環境下安裝不同的機 PyTorch 版本，虛擬環境支持互不影響。Ubuntu 環境下 anaconda 的安裝見博文 Ubuntu 之 Anaconda3 安裝。
wuhs@s169:~$ wget https://mirrors.bfsu.edu.cn/anaconda/archive/Anaconda3-2022.10-Linux-x86_64.sh
wuhs@s169:~$ sh Anaconda3-2022.10-Linux-x86_64.sh
wuhs@s169:~$ source ~/.bashrc
3、查看 Python 版本
emsp; emsp; 不同的 PyTorch 版本要求的 Python 版本是不一樣的，所以安裝好 anaconda3 后我們檢查當前的 Python 版本，默認初始化都是當前 anaconda3 發布時對應的 Python 最新版本，當然我們也可以使用 conda 創建所需的 Python 環境版本。我們在 torchvision 查看 PyTorch、torchvision、Python 版本匹配要求。
(base) wuhs@s169:~$ python -V
Python 3.9.13
4、安裝 PyTorch
emsp; emsp; 如下第二步，PyTorch 官網我們可以在選擇操作系統、安裝方式、編程語言、計算平臺后生成對應的安裝命令。
(base) wuhs@s169:~$ conda install pytorch torchvision torchaudio cpuonly -c pytorch
Collecting package metadata (current_repodata.json): done
Solving environment: done
## Package Plan ##
Proceed ([y]/n)? y
hellip;
5、版本驗證
(base) wuhs@s169:~$ python
Python 3.9.13 (main, Aug 25 2022, 23:26:10)
[GCC 11.2.0] :: Anaconda, Inc. on linux
Type“help”,“copyright”,“credits”or“license”for more information.
import torch
torch.version
lsquo;1.13.1 rsquo;
三、指定版本安裝 1、創建虛擬環境
(base) wuhs@s169:~$ conda create -n pytorch python=3.9
hellip;
(base) wuhs@s169:~$ conda activate pytorch
(pytorch) wuhs@s169:~$
2、安裝指定版本的 PyTorch
emsp; emsp; 安裝指定版本的 PyTorch 的時候我們需要在 GitHub 官網 PyTorch 頻道查看匹配版本，conda 安裝的時候指定版本號，具體版本號可以查看 anaconda 官網，PyTorch 版本和 TorchAudio 對應關系見
TorchAudio。當然如果我們指定版本錯誤的情況下，安裝的時候會報錯，我們根據報錯提示核驗是哪個軟件版本指定錯誤，再去官網合適確認修正后重新安裝即可。
(pytorch) wuhs@s169:~$ conda install pytorch3.12.0 torchvision=0.13.0 torchaudio0.12.0 cpuonly -c pytorch
到此，相信大家對“Linux 下怎么安裝 PyTorch”有了更深的了解，不妨來實際操作一番吧！這里是丸趣 TV 網站，更多相關內容可以進入相關頻道進行查詢，關注我們，繼續學習！
