Windows 11 に CUDA/cuDNN+Pytorch をインストールする

1. 準備作業：

torch のバージョンを確認します。Python インタラクティブ環境に入ります。

>>>import torch
>>>torch.__version__

cuda バージョンの確認: CMD ウィンドウ

nvcc --version

 バージョンが一致しない場合は、アンインストールして再インストールする必要があります。

2. インストール

Windows に CUDA/cuDNN をインストールする - Zhihu媒体 - Windows に CUDA をインストールする: 決定版ガイド媒体 - Windows 10 に CUDA と cuDNN をインストールする Windows は、cudn と cudnn のバージョン間の関係をインストールして構成します ... https://zhuanlan.zhihu.com/ p/99880204 CUDA と Pytorch 間の対応関係を表示します。

以前の PyTorch バージョン | PyTorch研究プロトタイピングから運用環境へのパスを加速するオープンソースの機械学習フレームワークhttps://pytorch.org/get-started/previous-versions/このマシンは CUDA11.8 なので、インストール コマンドは次のとおりです。最初の 3 つ:

# ROCM 5.4.2 (Linux only)
pip install torch==2.0.0+rocm5.4.2 torchvision==0.15.1+rocm5.4.2 torchaudio==2.0.1 --index-url https://download.pytorch.org/whl/rocm5.4.2
# CUDA 11.7
pip install torch==2.0.0+cu117 torchvision==0.15.1+cu117 torchaudio==2.0.1 --index-url https://download.pytorch.org/whl/cu117
# CUDA 11.8
pip install torch==2.0.0+cu118 torchvision==0.15.1+cu118 torchaudio==2.0.1 --index-url https://download.pytorch.org/whl/cu118
# CPU only
pip install torch==2.0.0+cpu torchvision==0.15.1+cpu torchaudio==2.0.1 --index-url https://download.pytorch.org/whl/cpu

3. 検証

pytorchに対応したcudaのバージョンを表示

python
>>>import torch
>>>torch.version.cuda

GPU が利用可能であることを確認する

import torch
import torchvision
print(torch.cuda.is_available())