POSTED BY
2023-05-19

WindowsにPythonとそのモジュールPyTorch、およびNVIDIAのCUDAツールキットを入れて、いざGPUを使おうとすると、

import torch
torch.cuda.is_available();
False

とあっさり否定されてGPUが使えない。

これは、Windowsのpipでノーオプションで普通に

pip install torch torchvision torchaudio

と、標準のものをインストールしてしまっていることが原因。オプションをつけて、専用サイトのものをインストールする必要がある。

まず既存のそれらを削除する。

pip uninstall torch torchvision torchaudio

そして、以下のように、インストールしているCUDAのバージョンに合ったURLを確認する。

https://download.pytorch.org/whl/cu118

ブラウザに上記を貼り付けて、ページが開けてモジュール一覧が表示されていればそのCUDAバージョン用のtorchが存在するということになる。当方11.8を入れているので、上記cu118となる。

確認できたらあらためてpipにオプションをつけてインストールする。

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

これで、晴れてPyTorchからCUDA-GPUが使用できるはずだ。

import torch
torch.cuda.is_available();
True