Windows10へのPyTorch 1.7.1+CUDA 11.0のインストール

2021年6月20日

バージョン組み合わせに注意

• pipの更新

  • python -m pip install -U pip setuptools

• CUDA Toolkit 11.0のインストール

  • https://developer.nvidia.com/cuda-11.0-download-archive

  • ダウンロードしたcuda_11.0.2_451.48_win10.exeを実行してインストール

• PyTorch 1.7.1のインストール

  • インストールコマンドを確認（https://pytorch.org/get-started/previous-versions/）
  • インストール

    pip install torch==1.7.1+cu110 torchvision==0.8.2+cu110 torchaudio==0.7.2 -f https://download.pytorch.org/whl/torch_stable.htmltorch_stable.html

• 動作確認

import torch
print(torch.rand(2, 3))