Linux下PyTorch安裝教學

2023-03-31 06:02:23

一、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、作業系統選擇

檢視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安裝

如上所示，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版本

不同的PyTorch版本要求的Python版本是不一樣的，所以安裝好anaconda3後我們檢查當前的Python版本，預設初始化都是當前anaconda3釋出時對應的Python最新版本，當然我們也可以使用conda建立所需的Python環境版本。我們在torchvision檢視PyTorch、torchvision、Python版本匹配要求。

(base) wuhs@s169:~$ python -V
Python 3.9.13

4、安裝PyTorch

如下第二步，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
…

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
‘1.13.1’
>>>

三、指定版本安裝

1、建立虛擬環境

(base) wuhs@s169:~$ conda create -n pytorch python=3.9
…
(base) wuhs@s169:~$ conda activate pytorch
(pytorch) wuhs@s169:~$

2、安裝指定版本的PyTorch

安裝指定版本的PyTorch的時候我們需要在GitHub官網PyTorch頻道檢視匹配版本，conda安裝的時候指定版本號，具體版本號可以檢視anaconda官網，PyTorch版本和TorchAudio對應關係見
TorchAudio。當然如果我們指定版本錯誤的情況下，安裝的時候會報錯，我們根據報錯提示核驗是哪個軟體版本指定錯誤，再去官網合適確認修正後重新安裝即可。