本篇內(nèi)容主要講解“Linux下怎么安裝PyTorch”，感興趣的朋友不妨來看看。本文介紹的方法操作簡單快捷，實用性強。下面就讓小編來帶大家學(xué)習(xí)“Linux下怎么安裝PyTorch”吧!

創(chuàng)新互聯(lián)堅持“要么做到，要么別承諾”的工作理念，服務(wù)領(lǐng)域包括：成都做網(wǎng)站、網(wǎng)站制作、企業(yè)官網(wǎng)、英文網(wǎng)站、手機端網(wǎng)站、網(wǎng)站推廣等服務(wù)，滿足客戶于互聯(lián)網(wǎng)時代的嵊泗網(wǎng)站設(shè)計、移動媒體設(shè)計的需求，幫助企業(yè)找到有效的互聯(lián)網(wǎng)解決方案。努力成為您成熟可靠的網(wǎng)絡(luò)建設(shè)合作伙伴！

一、PyTorch簡介

PyTorch是一個開源的Python機器學(xué)習(xí)庫，基于Torch，用于自然語言處理等應(yīng)用程序。2017年1月，由Facebook人工智能研究院（FAIR）基于Torch推出PyTorch。PyTorch的前身是Torch，其底層和Torch框架一樣，但是使用Python重新寫了很多內(nèi)容，不僅更加靈活，支持動態(tài)圖，而且提供了Python接口。它是由Torch7團(tuán)隊開發(fā)，是一個以Python優(yōu)先的深度學(xué)習(xí)框架，不僅能夠?qū)崿F(xiàn)強大的GPU加速，同時還支持動態(tài)神經(jīng)網(wǎng)絡(luò)。PyTorch既可以看作加入了GPU支持的numpy，同時也可以看成一個擁有自動求導(dǎo)功能的強大的深度神經(jīng)網(wǎng)絡(luò)。除了Facebook外，它已經(jīng)被Twitter、CMU和Salesforce等機構(gòu)采用。

二、安裝步驟

1、操作系統(tǒng)選擇

  查看PyTorch官網(wǎng)，可以看到PyTorch支持Linux、Mac、window平臺、支持conda、pip、源碼等安裝方式，也支持CPU、cuda、ROCm計算平臺，我們點擊環(huán)境選擇可以發(fā)現(xiàn)目前只有l(wèi)inux系統(tǒng)是支持全語言、全安裝方式、全計算平臺的，所以我們選擇linux操作系統(tǒng)作為系統(tǒng)環(huán)境。另外機器學(xué)習(xí)計算要求glibc版本要求較高，centos搭載的內(nèi)核和glbic版本較低，Ubuntu搭載的內(nèi)核版本都較新，所以機器學(xué)習(xí)主機建議使用Ubuntu操作系統(tǒng)。目前cuda更新支持的最低Ubuntu版本為18.04，所以建議使用Ubuntu18.04以上的操作系統(tǒng)。

wuhs@s169:~$ cat /etc/os-release
NAME=“Ubuntu”
VERSION=“18.04.6 LTS (Bionic Beaver)”

2、Anaconda3安裝

  如上所示，PyTorch支持的安裝方式有多種，博主擬采用conda安裝方式，建議先安裝Anaconda3，可以根據(jù)我們需要創(chuàng)建不同虛擬環(huán)境，虛擬環(huán)境下安裝不同的機PyTorch版本，虛擬環(huán)境支持互不影響。Ubuntu環(huán)境下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后我們檢查當(dāng)前的Python版本，默認(rèn)初始化都是當(dāng)前anaconda3發(fā)布時對應(yīng)的Python最新版本，當(dāng)然我們也可以使用conda創(chuàng)建所需的Python環(huán)境版本。我們在torchvision查看PyTorch、torchvision、Python版本匹配要求。

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

4、安裝PyTorch

  如下第二步，PyTorch官網(wǎng)我們可以在選擇操作系統(tǒng)、安裝方式、編程語言、計算平臺后生成對應(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、版本驗證

(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的時候我們需要在GitHub官網(wǎng)PyTorch頻道查看匹配版本，conda安裝的時候指定版本號，具體版本號可以查看anaconda官網(wǎng)，PyTorch版本和TorchAudio對應(yīng)關(guān)系見
TorchAudio。當(dāng)然如果我們指定版本錯誤的情況下，安裝的時候會報錯，我們根據(jù)報錯提示核驗是哪個軟件版本指定錯誤，再去官網(wǎng)合適確認(rèn)修正后重新安裝即可。

(pytorch) wuhs@s169:~$ conda install pytorch3.12.0 torchvision=0.13.0 torchaudio0.12.0 cpuonly -c pytorch

到此，相信大家對“Linux下怎么安裝PyTorch”有了更深的了解，不妨來實際操作一番吧！這里是創(chuàng)新互聯(lián)網(wǎng)站，更多相關(guān)內(nèi)容可以進(jìn)入相關(guān)頻道進(jìn)行查詢，關(guān)注我們，繼續(xù)學(xué)習(xí)！

當(dāng)前文章：Linux下怎么安裝PyTorch
當(dāng)前網(wǎng)址：http://chinadenli.net/article34/ppdspe.html

成都網(wǎng)站建設(shè)公司_創(chuàng)新互聯(lián)，為您提供Google、網(wǎng)站營銷、ChatGPT、自適應(yīng)網(wǎng)站、網(wǎng)站內(nèi)鏈、微信公眾號

廣告

聲明：本網(wǎng)站發(fā)布的內(nèi)容（圖片、視頻和文字）以用戶投稿、用戶轉(zhuǎn)載內(nèi)容為主，如果涉及侵權(quán)請盡快告知，我們將會在第一時間刪除。文章觀點不代表本網(wǎng)站立場，如需處理請聯(lián)系客服。電話：028-86922220；郵箱：631063699@qq.com。內(nèi)容未經(jīng)允許不得轉(zhuǎn)載，或轉(zhuǎn)載時需注明來源： 創(chuàng)新互聯(lián)