PyTorch インストール手順【環境別】徹底解説

PyTorchのインストールは、以下の手順で行うことができます:

  1. Pythonバージョンを確認してください:PyTorchはPython 3.6以上をサポートしています。Pythonバージョンを確認するには、ターミナルで次のコマンドを実行してください。
python --version
  1. pipの最新バージョンをインストールするには、次のコマンドを使用してpipをアップグレードできます:
pip install --upgrade pip
  1. PyTorchのインストール方法は、PyTorch公式ウェブサイトであなたのシステムに適したインストールコマンドを見つけることができます。たとえば、CUDA 10.2、Python 3.8、およびLinuxオペレーティングシステムを使用している場合、次のコマンドを使用してPyTorchをインストールできます。
pip install torch==1.7.0+cu102 torchvision==0.8.1+cu102 torchaudio==0.7.0 -f https://download.pytorch.org/whl/torch_stable.html
  1. Pythonのターミナルで、以下のコードを入力して、PyTorchのインストールが成功しているかを確認できます。
import torch
print(torch.__version__)

エラーが発生せず、PyTorchのバージョンが正常に表示された場合、PyTorchはシステムに正常にインストールされています。

bannerAds