torch1.7 のインストール#
1. cuda と cudnn のインストール#
詳細は以前に投稿した記事を参照してください。
2. 自分に適したバージョンを公式ウェブサイトで見つける#
Previous PyTorch Versions | PyTorch
私は cuda11 の Windows システムにインストールしています。以下のコマンドを使用します。異なるシステムではインストールコマンドが異なる場合がありますので、公式のコードを参照してください。
pip install torch==1.7.1+cu110 torchvision==0.8.2+cu110 torchaudio==0.7.2 -f https://download.pytorch.org/whl/torch_stable.html
エラーがなければインストールが完了です。インストールが遅い場合は pip のソースを変更してください。