🧩

PyTorch初心者挫折ポイント-PyTorchインストール編-

2023/10/19に公開

Python

今回はVSCode上でPyTorchのインストールする方法について解説します。
後半では、私がインストール時に躓いた点とその解決策を紹介します。

はじめに

PyTorchはオープンソースの機械学習ライブラリであり、特にニューラルネットワークの設計やトレーニング、評価を行う際に利用されます。
これから解説する方法はmac(M2)環境下におけるVSCode上でpipによるインストールを行う方法を取り扱います。
公式サイトによるダウンロードの一連の流れはこちら

インストール方法

PyTorch公式サイトにアクセス
INSTALL PYTORCHのセクションで自分の環境に合うような項目を選択
PyTorch ダウンロード選択
私の場合はこのように選択しました。
Run This Commandに表示されたpip3から始まる内容をコピーし、VSCode上で実行
以下のpythonコードを実行し、出力が得られることを確認

import torch
x = torch.rand(5, 3)
print(x)

# 出力
tensor([[0.3380, 0.3845, 0.3217],
        [0.8337, 0.9050, 0.2650],
        [0.2979, 0.7141, 0.9069],
        [0.1449, 0.1132, 0.1375],
        [0.4675, 0.3947, 0.1426]])

躓きポイント

% pip install torch
ERROR: Could not find a version that satisfies the requirement torch (from versions: none)
ERROR: No matching distribution found for torch

最初の段階で、私はこのようなエラーメッセージが発生してしまいました。
仮想環境の再構築や別のインストール方法を試しましたが改善は見られませんでした。

原因と解決策

この原因はPythonのバージョンがPyTorchのサポートしているバージョンではなかったことにありました。
PyTorchがサポートしているPythonのバージョンはPyTorchのリリースノートなどによって確認できます。今回ダウンロードするPyTorch(2.1.0)はPython3.11までしかサポートしていません。しかしながら、現在pythonはバージョン3.12.0までリリースされています。
私の場合も、python3.12.0の環境下でpytorch2.1.0をダウンロードしようとしたことに問題がありました。