首頁 > 運維 > linux運維 > Linux下PyTorch安裝的方法是什麼

Linux下PyTorch安裝的方法是什麼

PHPz
發布: 2023-05-14 14:25:13
轉載
4817 人瀏覽過

一、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以上的作業系統。

Linux下PyTorch安裝的方法是什麼

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版本匹配要求。

Linux下PyTorch安裝的方法是什麼

(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
Solvinging package metadata (current_repodata.json): done
Solving environment:ving environment done#.
## Package Plan

Linux下PyTorch安裝的方法是什麼

##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。當然如果我們指定版本錯誤的情況下,安裝的時候會報錯,我們根據報錯提示核驗是哪個軟體版本指定錯誤,再去官網合適確認修正後重新安裝即可。


Linux下PyTorch安裝的方法是什麼

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

#

以上是Linux下PyTorch安裝的方法是什麼的詳細內容。更多資訊請關注PHP中文網其他相關文章!

相關標籤:
來源:yisu.com
本網站聲明
本文內容由網友自願投稿,版權歸原作者所有。本站不承擔相應的法律責任。如發現涉嫌抄襲或侵權的內容,請聯絡admin@php.cn
熱門教學
更多>
最新下載
更多>
網站特效
網站源碼
網站素材
前端模板