Linux に PyTorch をインストールする方法

1. PyTorch の概要

PyTorch は、Torch に基づくオープンソースの Python 機械学習ライブラリであり、自然言語処理やその他のアプリケーションに使用されます。 2017 年 1 月、Facebook 人工知能研究所 (FAIR) によって、Torch に基づいて PyTorch が開始されました。 PyTorch の前身は、Torch です。その基礎となる層は、Torch フレームワークと同じですが、多くのコンテンツが Python で書き直されています。より柔軟で、ダイナミック グラフィックスをサポートしているだけでなく、Python インターフェイスも提供しています。これは、Torch7 チームによって開発された、強力な GPU アクセラレーションを可能にするだけでなく、動的ニューラル ネットワークもサポートする Python ファーストのディープ ラーニング フレームワークです。 PyTorch は、GPU サポートを備えた numpy とみなすこともできますが、自動導出機能を備えた強力なディープ ニューラル ネットワークとみなすこともできます。 Facebook に加えて、Twitter、CMU、Salesforce などの機関でも採用されています。

2. インストール手順

1. オペレーティング システムの選択

  PyTorch 公式 Web サイトを見ると、PyTorch が Linux、Mac、ウィンドウ プラットフォームをサポートしていることがわかります。 conda、pip やソース コードなどのインストール方法も CPU、cuda、ROCm コンピューティング プラットフォームをサポートしています。環境の選択をクリックすると、現在 Linux システムのみがすべての言語、すべてのインストール方法、すべてのコンピューティング プラットフォームをサポートしていることがわかります。 , そのため、システム環境として Linux オペレーティング システムを選択します。さらに、機械学習の計算にはより高いバージョンの glibc が必要であり、centos のカーネルお​​よび glibc バージョンはより低く、Ubuntu のカーネル バージョンは新しいため、機械学習ホストには Ubuntu オペレーティング システムを使用することをお勧めします。現在、cuda アップデートでサポートされている Ubuntu の最小バージョンは 18.04 であるため、Ubuntu 18.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 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. Python のバージョンを確認します

  異なる PyTorch のバージョンによって必要な Python のバージョンが異なるため、anaconda3 をインストールした後、現在の Python バージョンを確認します。デフォルトの初期化は、現在の anaconda3 リリースに対応する Python の最新バージョンです。もちろん、conda を使用して必要な Python 環境のバージョン。 torchvision では、PyTorch、torchvision、および Python バージョンの一致要件をチェックします。

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

4. PyTorch をインストールする

  2 番目のステップに示すように、PyTorch 公式 Web サイトで、オペレーティング システム、インストール方法、プログラミング言語、およびコンピューティング プラットフォームを選択した後、対応するインストール コマンドを生成できます。

(base) wuhs@s169:~$ conda install pytorch torchvision torchaudio cpuonly -c pytorch
パッケージ メタデータ (current_repodata.json) の収集: 完了
環境の解決: 完了

## パッケージ プラン

##続行 ([y]/n)? y
…

5. バージョンの確認

(base) wuhs@s169:~$ python
Python 3.9.13 (メイン、2022 年 8 月 25 日、23:26:10)
[GCC 11.2 .0] :: Linux 上の Anaconda, Inc.
詳細については、「ヘルプ」、「著作権」、「クレジット」または「ライセンス」と入力してください。
>>> import torch
> >> torch.version
‘1.13.1’
>>>

3. 指定したバージョンのインストール

1. 仮想環境を作成します

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

#2. 指定されたバージョンの PyTorch

をインストールします  指定されたバージョンの PyTorch
# をインストールする場合PyTorch の場合、GitHub 公式サイトの PyTorch チャネルでバージョンの一致を確認する必要があります。conda のインストール時にバージョン番号を指定します。具体的なバージョン番号は、anaconda 公式サイトで確認できます。PyTorch のバージョンと対応関係については、anaconda 公式サイトを参照してください。 TorchAudio、

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

以上がLinux に PyTorch をインストールする方法の詳細内容です。詳細については、PHP 中国語 Web サイトの他の関連記事を参照してください。