PyTorch開発環境の構築(ローカル編)

PyTorch環境の構築方法について説明します。

PyTorchの公式サイト(https://pytorch.org/)にアクセスします。

「Install」をクリックします。

自分の環境に合わせて選択します。選択肢に合わせて「Run this Command」にコマンドが表示されます。ここでは「Stable(1.11.0)」「Windows」「Pip」「Python」「CUDA11.3」を選択したところ、下記のコマンドが表示されました。

pip3 install torch torchvision torchaudio --extra-index-url https://download.pytorch.org/whl/cu113

PowerShellを開いてコマンドを入力し、実行すると、インストールが行われます。

PS C:\Users\user> pip3 install torch torchvision torchaudio --extra-index-url https://download.pytorch.org/whl/cu113
Looking in indexes: https://pypi.org/simple, https://download.pytorch.org/whl/cu113
Collecting torch
  Downloading https://download.pytorch.org/whl/cu113/torch-1.11.0%2Bcu113-cp310-cp310-win_amd64.whl (2186.0 MB)
     ━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━ 2.2/2.2 GB 2.3 MB/s eta 0:00:00
Collecting torchvision
  Downloading https://download.pytorch.org/whl/cu113/torchvision-0.12.0%2Bcu113-cp310-cp310-win_amd64.whl (5.4 MB)
     ━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━ 5.4/5.4 MB 48.8 MB/s eta 0:00:00
Collecting torchaudio
  Downloading https://download.pytorch.org/whl/cu113/torchaudio-0.11.0%2Bcu113-cp310-cp310-win_amd64.whl (573 kB)
     ━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━ 573.4/573.4 KB 17.6 MB/s eta 0:00:00
Collecting typing-extensions
  Downloading typing_extensions-4.2.0-py3-none-any.whl (24 kB)
Collecting requests
  Downloading requests-2.27.1-py2.py3-none-any.whl (63 kB)
     ━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━ 63.1/63.1 KB 1.7 MB/s eta 0:00:00
Collecting pillow!=8.3.*,>=5.3.0
  Downloading Pillow-9.1.1-cp310-cp310-win_amd64.whl (3.3 MB)
     ━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━ 3.3/3.3 MB 14.9 MB/s eta 0:00:00
Collecting numpy
  Downloading numpy-1.22.3-cp310-cp310-win_amd64.whl (14.7 MB)
     ━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━ 14.7/14.7 MB 29.8 MB/s eta 0:00:00
Collecting charset-normalizer~=2.0.0
  Downloading charset_normalizer-2.0.12-py3-none-any.whl (39 kB)
Requirement already satisfied: certifi>=2017.4.17 in c:\python310\lib\site-packages (from requests->torchvision) (2021.10.8)
Collecting urllib3<1.27,>=1.21.1
  Downloading urllib3-1.26.9-py2.py3-none-any.whl (138 kB)
     ━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━ 139.0/139.0 KB 4.0 MB/s eta 0:00:00
Requirement already satisfied: idna<4,>=2.5 in c:\python310\lib\site-packages (from requests->torchvision) (2.10)
Installing collected packages: urllib3, typing-extensions, pillow, numpy, charset-normalizer, torch, requests, torchvision, torchaudio
Successfully installed charset-normalizer-2.0.12 numpy-1.22.3 pillow-9.1.1 requests-2.27.1 torch-1.11.0+cu113 torchaudio-0.11.0+cu113 torchvision-0.12.0+cu113 typing-extensions-4.2.0 urllib3-1.26.9

インストールが完了したことを確認するため、PowerShellからPythonを起動し、「import torch」「torch.__version__」を順番に入力し、実行しましょう。正常にインストールされていれば、下記のようにPyTorchのバージョン(ここでは「’1.11.0+cu113’」)が表示されます。

PS C:\Users\user> python
Python 3.10.4 (tags/v3.10.4:9d38120, Mar 23 2022, 23:13:41) [MSC v.1929 64 bit (AMD64)] on win32
Type "help", "copyright", "credits" or "license" for more information.
>>> import torch
>>> torch.__version__
'1.11.0+cu113'
>>> quit()