.jpg)
一、pytorch簡(jiǎn)介
pytorch是一個(gè)開(kāi)源的python機(jī)器學(xué)習(xí)庫(kù)，基于torch，用于自然語(yǔ)言處理等應(yīng)用程序。2017年1月，由facebook人工智能研究院（fair）基于torch推出pytorch。pytorch的前身是torch，其底層和torch框架一樣，但是使用python重新寫(xiě)了很多內(nèi)容，不僅更加靈活，支持動(dòng)態(tài)圖，而且提供了python接口。它是由torch7團(tuán)隊(duì)開(kāi)發(fā)，是一個(gè)以python優(yōu)先的深度學(xué)習(xí)框架，不僅能夠?qū)崿F(xiàn)強(qiáng)大的gpu加速，同時(shí)還支持動(dòng)態(tài)神經(jīng)網(wǎng)絡(luò)。pytorch既可以看作加入了gpu支持的numpy，同時(shí)也可以看成一個(gè)擁有自動(dòng)求導(dǎo)功能的強(qiáng)大的深度神經(jīng)網(wǎng)絡(luò)。除了facebook外，它已經(jīng)被twitter、cmu和salesforce等機(jī)構(gòu)采用。
二、安裝步驟
1、操作系統(tǒng)選擇
??查看PyTorch官網(wǎng)，可以看到PyTorch支持linux、Mac、window平臺(tái)、支持conda、pip、源碼等安裝方式，也支持CPU、cuda、ROCm計(jì)算平臺(tái)，我們點(diǎn)擊環(huán)境選擇可以發(fā)現(xiàn)目前只有linux系統(tǒng)是支持全語(yǔ)言、全安裝方式、全計(jì)算平臺(tái)的，所以我們選擇linux操作系統(tǒng)作為系統(tǒng)環(huán)境。另外機(jī)器學(xué)習(xí)計(jì)算要求glibc版本要求較高，centos搭載的內(nèi)核和glbic版本較低，ubuntu搭載的內(nèi)核版本都較新，所以機(jī)器學(xué)習(xí)主機(jī)建議使用Ubuntu操作系統(tǒng)。目前cuda更新支持的最低Ubuntu版本為18.04，所以建議使用Ubuntu18.04以上的操作系統(tǒng)。
wuhs@s169:~$ cat /etc/os-releaseNAME=“Ubuntu”VERSION=“18.04.6 LTS (Bionic Beaver)”
2、Anaconda3安裝
??如上所示，PyTorch支持的安裝方式有多種，博主擬采用conda安裝方式，建議先安裝Anaconda3，可以根據(jù)我們需要?jiǎng)?chuàng)建不同虛擬環(huán)境，虛擬環(huán)境下安裝不同的機(jī)PyTorch版本，虛擬環(huán)境支持互不影響。Ubuntu環(huán)境下anaconda的安裝見(jiàn)博文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后我們檢查當(dāng)前的Python版本，默認(rèn)初始化都是當(dāng)前anaconda3發(fā)布時(shí)對(duì)應(yīng)的Python最新版本，當(dāng)然我們也可以使用conda創(chuàng)建所需的Python環(huán)境版本。我們?cè)趖orchvision查看PyTorch、torchvision、Python版本匹配要求。
(base) wuhs@s169:~$ python -V
Python 3.9.13
4、安裝PyTorch
??如下第二步，PyTorch官網(wǎng)我們可以在選擇操作系統(tǒng)、安裝方式、編程語(yǔ)言、計(jì)算平臺(tái)后生成對(duì)應(yīng)的安裝命令。
(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、版本驗(yàn)證
(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 “l(fā)icense” for more information.
>>> import torch
>>> torch.version
‘1.13.1’
>>>
三、指定版本安裝
1、創(chuàng)建虛擬環(huán)境
(base) wuhs@s169:~$ conda create -n pytorch python=3.9
…
(base) wuhs@s169:~$ conda activate pytorch
(pytorch) wuhs@s169:~$
2、安裝指定版本的PyTorch
??安裝指定版本的PyTorch的時(shí)候我們需要在github官網(wǎng)PyTorch頻道查看匹配版本，conda安裝的時(shí)候指定版本號(hào)，具體版本號(hào)可以查看anaconda官網(wǎng)，PyTorch版本和TorchAudio對(duì)應(yīng)關(guān)系見(jiàn)
TorchAudio。當(dāng)然如果我們指定版本錯(cuò)誤的情況下，安裝的時(shí)候會(huì)報(bào)錯(cuò)，我們根據(jù)報(bào)錯(cuò)提示核驗(yàn)是哪個(gè)軟件版本指定錯(cuò)誤，再去官網(wǎng)合適確認(rèn)修正后重新安裝即可。
(pytorch) wuhs@s169:~$ conda install pytorch2.12.0 torchvision=0.13.0 torchaudio0.12.0 cpuonly -c pytorch
.png)
