Windows11にPython3.10.8+CUDA11.6+cuDNN8.5+PyTorch1.12.1(GPU版:cu116)環境作成

AnacondaとかMinicondaとか使わずpipで環境構築した。

環境確認

import torch
print(torch.__version__)
print(torch.cuda.is_available())
print(torch.cuda.device_count())
print(torch.version.cuda)

結果

$ python a.py
1.12.1+cu116
True
1
11.6

$ python -V
Python 3.10.8

参考

CUDA+cuDNNをインストールしPyTorchでGPUを認識させるまでの手順(Window11)

• https://zenn.dev/ryu2021/articles/3d5737408b06fe

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

感想

CUDAとかcuDNNとかPyTorchの知見が全くない中でWindows11に環境構築するのはなかなか難しい。最初から最新のPyTorch(GPU版)に対応するCUDAをインストールしていたら詰まることはなかった。