So installieren Sie PyTorch unter Linux

1. Einführung in PyTorch

PyTorch ist eine Open-Source-Python-Bibliothek für maschinelles Lernen, die auf Torch basiert und für die Verarbeitung natürlicher Sprache und andere Anwendungen verwendet wird. Im Januar 2017 wurde PyTorch auf Basis von Torch vom Facebook Artificial Intelligence Research Institute (FAIR) eingeführt. Der Vorgänger von PyTorch ist Torch. Seine zugrunde liegende Ebene ist die gleiche wie das Torch-Framework, viele Inhalte wurden jedoch in Python neu geschrieben. Es ist nicht nur flexibler, unterstützt dynamische Grafiken, sondern bietet auch eine Python-Schnittstelle. Es wurde vom Torch7-Team entwickelt und ist ein Python-First-Deep-Learning-Framework, das nicht nur eine leistungsstarke GPU-Beschleunigung ermöglicht, sondern auch dynamische neuronale Netze unterstützt. PyTorch kann als Numpy mit GPU-Unterstützung und auch als leistungsstarkes tiefes neuronales Netzwerk mit automatischer Ableitungsfunktion angesehen werden. Neben Facebook wurde es von Institutionen wie Twitter, CMU und Salesforce übernommen.

2. Installationsschritte

1. Auf der offiziellen Website von PyTorch können Sie sehen, dass PyTorch Linux, Mac, Windows-Plattformen und andere Installationsmethoden unterstützt. und unterstützt auch die Computerplattformen CPU, Cuda und ROCm. Wenn wir auf die Umgebungsauswahl klicken, können wir feststellen, dass derzeit nur das Linux-System alle Sprachen, alle Installationsmethoden und alle Computerplattformen unterstützt. Daher wählen wir das Linux-Betriebssystem als Systemumgebung. Darüber hinaus erfordern Berechnungen für maschinelles Lernen eine höhere Version von glibc, die Kernel- und glbic-Versionen von Centos sind niedriger und die Kernel-Versionen von Ubuntu sind neuer. Daher wird empfohlen, das Ubuntu-Betriebssystem für Hosts für maschinelles Lernen zu verwenden. Derzeit ist die vom cuda-Update unterstützte Mindestversion von Ubuntu 18.04, daher wird empfohlen, ein Betriebssystem von Ubuntu 18.04 oder höher zu verwenden.

wuhs@s169:~$ cat /etc/os-release

NAME="Ubuntu"

VERSION="18.04.6 LTS (Bionic Beaver)"

2, Anaconda3-Installation

 &emsp ; Wie oben gezeigt, unterstützt PyTorch verschiedene Installationsmethoden. Es wird empfohlen, zuerst Anaconda3 zu installieren virtuelle Umgebung. Die virtuelle Umgebung unterstützt sich gegenseitig. Informationen zur Installation von Anaconda in der Ubuntu-Umgebung finden Sie im Blog-Beitrag Ubuntu Anaconda3 Installation.

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. Überprüfen Sie die Python-Version

   Verschiedene PyTorch-Versionen erfordern unterschiedliche Python-Versionen. Daher überprüfen wir nach der Installation von anaconda3 die aktuelle Python-Version. Die Standardinitialisierung ist die neueste Version von Python, die der aktuellen Version von Anaconda3 entspricht. Natürlich können wir auch Conda verwenden, um die erforderliche Python-Umgebungsversion zu erstellen. Bei Torchvision prüfen wir die Übereinstimmungsanforderungen für PyTorch, Torchvision und Python-Versionen.

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

Python 3.9.13

4. Installieren Sie PyTorch

  Der zweite Schritt ist wie folgt. Wir können das Betriebssystem auswählen Auf der offiziellen PyTorch-Website wird der entsprechende Installationsbefehl generiert, nachdem die Installationsmethode, die Programmiersprache und die Computerplattform festgelegt wurden. ??

Weiter ([j]/n)? 10)

[GCC 11.2.0] :: Anaconda, Inc. unter Linux

Geben Sie „Hilfe“, „Copyright“, „Credits“ oder „Lizenz“ ein, um weitere Informationen zu erhalten.

>>> Import Torch

> >> Torch.
Version

‘1.13.1’
>>>

3. Installation der angegebenen Version

1. Erstellen Sie eine virtuelle Umgebung

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

…

(base) wuhs@s169:~$ conda activate pytorch

(pytorch) wuhs@s169:~$

2. Installieren Sie die angegebene Version von PyTorch

&emsp ;  Bei der Installation der angegebenen Version von PyTorch müssen wir die entsprechende Version im PyTorch-Kanal der offiziellen Website von GitHub überprüfen. Bei der Installation von Conda können Sie die spezifische Versionsnummer überprüfen . Die entsprechende Beziehung zwischen der PyTorch-Version und TorchAudio finden Sie unter
TorchAudio. Wenn wir die falsche Version angeben, wird während der Installation natürlich ein Fehler gemeldet. Wir können anhand der Fehlermeldung überprüfen, welche Softwareversion falsch ist, und dann auf die offizielle Website gehen, um die Korrektur zu bestätigen und neu zu installieren.

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

Das obige ist der detaillierte Inhalt vonSo installieren Sie PyTorch unter Linux. Für weitere Informationen folgen Sie bitte anderen verwandten Artikeln auf der PHP chinesischen Website!